并查集 AcWing 240. 食物链
原题链接
算法标签
并查集
思路
用并查集维护每条食物链之间的关系, 对于每类动物之间, 采用该动物距离根节点之间的距离来表示动物之间的关系
具体的见下图
对于该并查集 需要额外维护距离
代码即
if(p[x]!=x){
// u为祖宗节点
int u=find(p[x]);
// 更新 当前点距祖宗节点距离 = 当前点距父节点距离 + 父节点距祖宗节点距离
d[x]+=d[p[x]];
// 更新 当前点父亲节点 = 祖宗节点
p[x]=u;
}
对于同类关系
若不属于同一条食物链
需要进行并查集合并
合并时新边权值为
对于捕食关系
若不属于同一条食物链
需要进行并查集合并
合并时新边权值为
具体思路见代码
代码
#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 100005;
// p存储节点祖宗节点 d存储当前节点据祖宗节点的距离
int p[N], d[N];
inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
void put(int x) {
if(x<0) putchar('-'),x=-x;
if(x>=10) put(x/10);
putchar(x%10^48);
}
// 查询祖宗节点
int find(int x){
if(p[x]!=x){
// u为祖宗节点
int u=find(p[x]);
// 更新 当前点距祖宗节点距离 = 当前点距父节点距离 + 父节点距祖宗节点距离
d[x]+=d[p[x]];
// 更新 当前点父亲节点 = 祖宗节点
p[x]=u;
}
return p[x];
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n=read(), m=read();
rep(i, 1, n+1){
p[i]=i;
}
int res=0;
while(m--){
int t=read(), x=read(), y=read();
if(x>n||y>n){
res++;
}else{
int px=find(x), py=find(y);
if(t==1){
// 如果在一条食物链且距离 % 3为1, 2 表明不是同类
if(px==py&&(d[x]-d[y])%3){
res++;
}else if(px!=py){/* 如果不在同一条食物链
进行食物链合并
对于合并边的权值为
为保证x, y属于同一类
需要使(x到py的距离 + 合并边的权值 - y到py的距离) % 3 = 0
即x到py的距离 = y到py的距离 */
p[px]=py;
d[px]=d[y]-d[x];
}
}else{
// 如果在一条食物链且距离 - 1 % 3为1, 2 表明不是捕食关系
if(px==py&&(d[x]-d[y]-1)%3){
res++;
}else if(px!=py){/* 如果不在同一条食物链
进行食物链合并
对于合并边的权值为
为保证x, y属于为捕食关系
需要使(x到py的距离 + 合并边的权值 - y到py的距离 - 1) % 3 = 0*/
p[px]=py;
d[px]=d[y]+1-d[x];
}
}
}
}
printf("%lld", res);
return 0;
}
原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈