再次做经典题
day1 打卡
方法一 带权并查集
刚写完上一题,很自然的就想到用带权并查集解决问题,设x与y为给定的两种动物,可以想到有两种关系:一是x与y之前已经确定过关系,二是x与y之前没有确定过关系。
如果是第一种情况,我们根据权值确定之前记录的关系是否满足现在给定的关系,如果不满足假话数加一。
如果是第二种情况,我们可以将x与y进行合并,确定一个为另一个的父亲节点,同时更新w[x],w[y]中的权值。我们可以想象合并后的x与y都有一个共同的根(祖先)节点,计算x与y到根节点的差值对三取模,即为x与y的关系,总共有三种情况,我们可以规定0为同级,1为x吃y,2为y吃x。
(我自己有一步逻辑上的错误,WA了好几发,就是有两个条件,一是给定的值大于n,二是在谁吃谁的时候x与y的值相等,一开始我把这两个用if语句单独列出来判断,导致最后的假话数量被多加了几次,后来才反应过来,tcl)
AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50010;
int f[maxn];
int w[maxn];
int ans=0;
int find(int x){
if(x==f[x]) return x;
int fx=f[x];
f[x]=find(f[x]);
w[x]=w[x]+w[fx];
return f[x];
}
void add(int x,int y,int val){
int xx=find(x);
int yy=find(y);
if(xx!=yy) f[xx]=yy;
w[xx]=val+w[y]-w[x];
}
int main(){
int n,k;
scanf("%d%d",&n,&k);
for(int i=0;i<=n;i++) f[i]=i;
while(k--){
int opp,x,y;
scanf("%d%d%d",&opp,&x,&y);
if(x>n||y>n) ans++;
else{
int xx=find(x);
int yy=find(y);
if(opp==1){
if(xx!=yy){
add(x,y,3);
}else{
if((w[x]-w[y])%3!=0) ans++;
}
}else{
if(x==y) ans++;
else{
if(xx!=yy){
add(x,y,1);
}else{
if(abs(w[x]-w[y]-1)%3!=0) ans++;
}
}
}
}
}
printf("%d",ans);
return 0;
}
方法二 种类并查集
搜到的大佬讲解,一下就明白了种类并查集为什么要开关系倍的数组:种类并查集讲解
种类并查集即是有多少层关系,就用多少倍的数组存这些关系,如本题一种动物和其它动物可以有三种关系,分别是同类,捕食,天敌,我们的数组就开三倍的大小,去储存这些关系,然后判断。
AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50010;
int f[3*maxn];
int ans=0;
int find(int x){
while(x!=f[x]) x=f[x];
return x;
}
int main(){
int n,k;
scanf("%d%d",&n,&k);
for(int i=0;i<=3*n;i++) f[i]=i;
while(k--){
int opp,x,y;
scanf("%d%d%d",&opp,&x,&y);
if(x>n||y>n){
ans++;
continue;
}
if(opp==1){
if(find(x+n)==find(y)||find(x+2*n)==find(y)){
ans++;
continue;
}
f[find(x)]=find(y);
f[find(x+n)]=find(y+n);
f[find(x+2*n)]=find(y+2*n);
}else if(opp==2){
if(find(x)==find(y)||find(x+2*n)==find(y)){
ans++;
continue;
}
f[find(x)]=find(y+2*n);
f[find(x+n)]=find(y);
f[find(x+2*n)]=find(y+n);
}
}
printf("%d",ans);
return 0;
}