依然并查集。。。但是。。。这题一晚上居然没调出来
本来是想通过不压缩状态实现,但是超时,又想到通过每个点的直接父节点推导,结果发现自己推的式子错了
此路不通,实在受不了去搜了标程。。。。这你妹的
各个函数都改成标程的功能了,居然wa,求各位大大找茬。。。。
先标程
#include<iostream>
#include<cstdio>
using namespace std;
struct node{
int up,count,parent;
};
node Cu[30001];
void creat(){
int c;
for(c=0;c<=30000;c++){
Cu[c].count=1;
Cu[c].parent=c;
Cu[c].up=0;
}
}
int find_set(int i){
int t=Cu[i].parent;
if(t!=i){
Cu[i].parent=find_set(t);
Cu[i].up+=Cu[t].up;
}
return Cu[i].parent;
}
void uion(int i,int j){
int x=find_set(i);
int y=find_set(j);
Cu[y].parent=x;
Cu[y].up=Cu[x].count;
Cu[x].count+=Cu[y].count;
}
int main(){
creat();
int n,a,b;
char s;
cin>>n;
while(n--){
cin>>s;
if(s=='M'){
scanf("%d %d",&a,&b);
uion(a,b);
}
else if(s=='C'){
scanf("%d",&a);
b=find_set(a);
printf("%d\n",Cu[b].count-Cu[a].up-1);
}
}
return 0;
}
然后是我自己的。。。
#include<iostream>
#include<cstdio>
using namespace std;
int x,y,p,sets[50000],val[50000],up[50000];
char c;
//bool flag[50000];
int find(int x){
int y=sets[x];
if(sets[x]==x) return x;
else{
up[x]+=up[y];
y=find(sets[x]);
sets[x]=y;
return y;
}
}
void move(int x,int y){
int a,b;
a=find(x);
b=find(y);
//cout<<"moves "<<a<<" to "<<b<<endl;
//flag[a]=true;
sets[b]=a;
up[b]=val[a];
val[a]+=val[b];
}
int cnt(int x){
int y=find(x);
cout<<val[y]-up[x]-1<<endl;
}
int main(){
for(int i=0;i<50000;i++){
sets[i]=i;
val[i]=1;
up[i]=0;
//flag[i]=false;
}
scanf("%d",&p);
while(p--){
/*for(int i=1;i<=6;i++)
cout<<i<<"belong to"<<sets[i]<<" val="<<val[i]<<" UP="<<up[i]<<endl;
cout<<endl;*/
//cout<<p<<endl;
getchar();
scanf("%c",&c);
if(c=='C'){
scanf("%d",&x);
cnt(x);
}
else if(c=='M'){
scanf("%d%d",&x,&y);
move(x,y);
}
}
return 0;
}
泪奔中。。。勿扰