cpp
#include<bits/stdc++.h>
using namespace std;
const int N=50000+10;
int p[N],d[N];
int find(int x)
{
if(x!=p[x])
{
int u=find(p[x]);
d[x]+=d[p[x]];
p[x]=u;
}
return p[x];
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
int ans=0;
while(m--)
{
int t,x,y;
cin>>t>>x>>y;
if(x>n||y>n)
{
ans++;
continue;
}
int px=find(x),py=find(y);
if(t==1)
{
if(px==py&&(d[x]-d[y])%3) ans++;
else if(px!=py)
{
p[px]=py;
d[px]=d[y]-d[x];
}
}
else
{
if(px==py&&(d[x]-d[y]-1)%3) ans++;
else if(px!=py)
{
p[px]=py;
d[px]=d[y]+1-d[x];
}
}
}
cout<<ans<<endl;
return 0;
}
使用并查集维护一些额外信息,每一个节点到根节点的距离对3取模,表示该节点和根节点的关系,假设取模之后的结果是0,表示和根节点是同类,假设结果是1,表示可以吃根节点,结果是2,表示可以被根节点吃,只有三种情况,同类,吃根节点,被根节点吃,同类之间不可以互相吃
函数调用的时候不可以多次调用,否则会发生错误
要知道两个生物之间的关系,可以根据他们和根节点之间的关系来判断,比如说,一个到根节点的距离是1(x
),一个到根节点的距离是2(y
),那么 y
吃x
情况有多种,没有穷举完所有情况不要使用 else
假设单次修改会改变我们后面要使用的变量,那么我们用一个中间变量(或者叫临时变量)把将被修改的变量存下来,例如
cpp
int u=find(p[x]);
d[x]+=d[p[x]];
p[x]=u;
长度的变化可以画示意图来进行理解
p
存的是父节点,d
存的是到根节点的距离,只保证到根节点的距离是有意义的,这样子可以保证时间复杂度是O(n)