【并查集】【模拟】Codeforces 698B & 699D Fix a Tree

最新推荐文章于 2022-04-26 22:50:14 发布

转载最新推荐文章于 2022-04-26 22:50:14 发布 · 74 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Coolxxx/p/5786051.html

本文解析了 Codeforces 平台上的两道题目，详细介绍了如何使用并查集算法来解决树结构的问题，包括错误树的修正和环的检测。通过实际代码展示了算法的具体实现过程。

题目链接：

　　http://codeforces.com/problemset/problem/698/B

　　http://codeforces.com/problemset/problem/699/D

题目大意：

　　通过给定当前节点的父亲给你一棵有错的树，可能有多个根和环，输出改成正确的一棵树至少要修改几个节点的父亲和修改后所有点的父亲值

题目思路：

　　【并查集】【模拟】

　　用并查集把成环的归在一起(类似强连通分量)，然后统计分量数并修改。

　　第一个出现的当作根，其余的每一块连通分量都去掉一条边改为连接到根上。

 1 //
 2 //by coolxxx
 3 ////<bits/stdc++.h>
 4 #include<iostream>
 5 #include<algorithm>
 6 #include<string>
 7 #include<iomanip>
 8 #include<map>
 9 #include<memory.h>
10 #include<time.h>
11 #include<stdio.h>
12 #include<stdlib.h>
13 #include<string.h>
14 //#include<stdbool.h>
15 #include<math.h>
16 #define min(a,b) ((a)<(b)?(a):(b))
17 #define max(a,b) ((a)>(b)?(a):(b))
18 #define abs(a) ((a)>0?(a):(-(a)))
19 #define lowbit(a) (a&(-a))
20 #define sqr(a) ((a)*(a))
21 #define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
22 #define mem(a,b) memset(a,b,sizeof(a))
23 #define eps (1e-8)
24 #define J 10
25 #define MAX 0x7f7f7f7f
26 #define PI 3.14159265358979323
27 #define N 200004
28 using namespace std;
29 typedef long long LL;
30 int cas,cass;
31 int n,m,lll,ans;
32 int root;
33 int a[N],fa[N];
34 int zhao(int aa)
35 {
36     if(fa[aa]==0 || fa[aa]==aa)return aa;
37     return (fa[aa]=zhao(fa[aa]));
38 }
39 int main()
40 {
41     #ifndef ONLINE_JUDGE
42     freopen("1.txt","r",stdin);
43 //    freopen("2.txt","w",stdout);
44     #endif
45     int i,j,fx,fy;
46 //    for(scanf("%d",&cas);cas;cas--)
47 //    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
48     while(~scanf("%d",&n))
49 //    while(~scanf("%d",&n))
50     {
51         mem(fa,0);ans=0;root=0;
52         for(i=1;i<=n;i++)
53         {
54             scanf("%d",&a[i]);
55             if(a[i]==i)root=i;
56             fx=zhao(a[i]);
57             fy=zhao(i);
58             if(fx!=fy)
59                 fa[fy]=fx;
60         }
61         for(i=1;i<=n;i++)
62         {
63             fa[i]=zhao(i);
64             if(fa[i]==i)
65             {
66                 if(!root)
67                 {
68                     ans++;
69                     root=i;
70                     a[i]=i;
71                 }
72                 else if(i!=root)
73                     ans++,a[i]=root;
74             }
75         }
76         printf("%d\n",ans);
77         for(i=1;i<=n;i++)
78             printf("%d ",a[i]);
79         puts("");
80     }
81     return 0;
82 }
83 /*
84 //
85 
86 //
87 */