首页 > 代码库 > 【并查集&&带权并查集】BZOJ3296&&POJ1182
【并查集&&带权并查集】BZOJ3296&&POJ1182
bzoj1529[POI2005]ska Piggy banks
【题目大意】
n头奶牛m种语言,每种奶牛分别掌握一些语言。问至少再让奶牛多学多少种语言,才能使得它们能够直接或间接交流?
【思路】
(n+m)个点,奶牛学会某种语言就合并它和语言的节点。并查集维护联通块,答案为联通块个数-1。水,可是我跳坑了。
我一开始做法是设总的联通块有(n+m)个,然后没合并一次减去1。其实这样是不可行的,因为我们只需要考虑奶牛(即节点1..n)有几个联通块。有可能一些语言根本没有任何奶牛掌握……
1 #include<bits/stdc++.h> 2 using namespace std; 3 const int MAXN=10000+30000+50; 4 int n,m,t; 5 int u[MAXN],h[MAXN],appear[MAXN]; 6 7 int find(int x) 8 { 9 int tmp=x; 10 while (u[tmp]!=tmp) tmp=u[tmp]; 11 while (u[x]!=x) 12 { 13 int temp=u[x]; 14 u[x]=tmp; 15 x=temp; 16 } 17 return tmp; 18 } 19 20 void union_set(int fa,int fb) 21 { 22 if (h[fa]>=h[fb]) 23 { 24 u[fb]=fa; 25 if (h[fa]==h[fb]) h[fa]++; 26 } 27 else u[fa]=fb; 28 } 29 30 void solve() 31 { 32 scanf("%d%d",&n,&m); 33 for (int i=1;i<=(n+m);i++) u[i]=i,h[i]=0; 34 for (int i=1;i<=n;i++) 35 { 36 int k,lang; 37 scanf("%d",&k); 38 for (int j=1;j<=k;j++) 39 { 40 scanf("%d",&lang); 41 int fa=find(i),fb=find(lang+n); 42 if (fa!=fb) union_set(fa,fb); 43 } 44 } 45 int t=0; 46 memset(appear,0,sizeof(appear)); 47 for (int i=1;i<=n;i++) 48 { 49 int fa=find(i); 50 if (appear[fa]==0) t++,appear[fa]=1; 51 } 52 printf("%d",t-1); 53 } 54 55 int main() 56 { 57 solve(); 58 return 0; 59 }
POJ1182-[NOI01]食物链
【题目大意】
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
【思路】
*我上一次写这道题是在一年半之前。哦。
用0表示当前节点和u[i]同类,1表示当前节点i吃u[i],2表示u[i]吃当前节点i。
我们设一个v,对于题目中的输入,当为1时,v=0,当为2时,v=1。
显然如果要三个节点a、b、c,其中u[b]=c,u[a]=b,如果要压缩为u[a]=c,那么rank[a]=(rank[a]+rank[b])%3。
然后其他推一推,详见程序吧..
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 const int MAXN=50000+50; 8 int rank[MAXN],u[MAXN]; 9 int n,k,ans; 10 11 int find(int x) 12 { 13 if (x!=u[x]) 14 { 15 int tmp=u[x]; 16 u[x]=find(u[x]); 17 rank[x]=(rank[x]+rank[tmp])%3; 18 } 19 return u[x]; 20 } 21 22 void init() 23 { 24 ans=0; 25 scanf("%d%d",&n,&k); 26 for (int i=1;i<=n;i++) rank[i]=0,u[i]=i; 27 } 28 29 void solve() 30 { 31 for (int i=1;i<=k;i++) 32 { 33 int d,x,y; 34 scanf("%d%d%d",&d,&x,&y); 35 if (x>n||y>n||((d==2) && (x==y))) 36 { 37 ans++; 38 continue; 39 } 40 int fx=find(x),fy=find(y); 41 int v=(d==1)?0:1; 42 if (fx!=fy) 43 { 44 u[fx]=fy; 45 rank[fx]=(rank[y]+v-rank[x]+3)%3; 46 } 47 else 48 if (rank[x]!=(rank[y]+v)%3) ans++; 49 } 50 printf("%d",ans); 51 } 52 53 int main() 54 { 55 init(); 56 solve(); 57 return 0; 58 }
【并查集&&带权并查集】BZOJ3296&&POJ1182