http://poj.org/problem?id=1182
Time Limit: 1000MS | Memory Limit: 10000K | |
Total Submissions: 44998 | Accepted: 13122 |
Description
Input
Output
Sample Input
100 7 1 101 1 2 1 2 2 2 3 2 3 3 1 1 3 2 3 1 1 5 5
Sample Output
3分析:该题是种类并查集,每个动物都属于A,B,C三类中的一类,始终满足A<-吃-B<-吃-C<-吃-A的关系;
用两个数组即可,f[]表示祖先节点,mark[]表示每个节点的状态
所以用0,1,2,三个数表示与祖先节点的关系:0表示与祖先同类,1表示被祖先吃,2表示吃祖先;
举个例子就很清楚了
设被吃者指向父节点即吃者
(1)1吃2:
A=1的祖先是1此时x=f[1]=1, mark[1]=0;
B=2的祖先是2此时y=f[2]=2], mark[2]=0;
因为1吃2,所以f[y]=x即:f[2]=1; 2的状态比1 大1,
所以mark[y]=(mark[a]+1-mark[b]+3)%3;(就是把当前的增量平移到祖先身上)
==>mark[1]=0;mark[2]=1;
(2) 2吃3;
A=2的祖先是1此时x=f[2]=1;mark[2]=1;
B=3的祖先是3此时y=f[3]=3; mark[3]=0;
f[y]=x;即:f[3]=1;
所以mark[y]=(mark[a]+1-mark[b]+3)%3;
==>mark[3]=2;
此时图形是:
(3)同理还有4吃5,5吃6可构成下图:
(4)此时加入3吃5:
A=3,x=f[3]=1;mark[3]=2;
B=5;y=f[5]=4;mark[5]=1;
3的状态要比5的状态小1,所以看看要是把5集合加入到3的集合中,5的祖先平移 量是多少mark[y]=(mark[a]+1-mark[b]+3)%3,因为mark[y]在任何时候都是mark[y]=0;所以直接赋值就行了,此时:
此时只是更新了祖先的mark值;
当找5的时候务必会进行一次finde()函数:
int finde(int x)
{
if(x!=f[x])
{
int t=f[x];
f[x]=finde(f[x]);
mark[x]=(mark[x]+mark[t])%3;
}
return f[x];
}
F[5]=1;mark[5]=0;
同理要求6的话f[6]=1;mark[6]=1;
可以看出mark值相同的动物是一类的;
加入3和5是同类的话
Mark[y]=(mark[a]-mark[b]+3)%3;
Mark[4]=1;mark[5]=2;mark[6]=0;
程序:
#include"string.h" #include"stdio.h" #include"queue" #include"stack" #include"iostream" #include"math.h" #define M 60006 #define inf 100000000 using namespace std; int f[M],mark[M]; int finde(int x) { if(x!=f[x]) { int t=f[x]; f[x]=finde(f[x]); mark[x]=(mark[x]+mark[t])%3; } return f[x]; } void make(int a,int b) { int x=finde(a); int y=finde(b); if(x!=y) { f[y]=x; mark[y]=(3+mark[a]+1-mark[b])%3; } } void same(int a,int b) { int x=finde(a); int y=finde(b); if(x!=y) { f[y]=x; mark[y]=(3+mark[a]-mark[b])%3; } } int main() { int n,m,i; scanf("%d%d",&n,&m); for(i=1;i<=n;i++) f[i]=i; memset(mark,0,sizeof(mark)); int wrong=0; while(m--) { int d,a,b; scanf("%d%d%d",&d,&a,&b); if(a>n||b>n||a<1||b<1) { wrong++; //printf("Wrong\n"); continue; } int x=finde(a); int y=finde(b); if(d==1) { if(x!=y) same(a,b); else { if(mark[a]!=mark[b]) { //printf("Wrong\n"); wrong++; } } } if(d==2) { if(x!=y) make(a,b); else { if(mark[a]+1!=mark[b]&&mark[a]!=mark[b]+2) { //printf("Wrong\n"); wrong++; } } } } printf("%d\n",wrong); return 0; }