传送门:http://poj.org/problem?id=1182
这题真是……我尼玛啊……弄到我吐血。其实如果搞懂了原理的话,也就是几行核心代码的事儿。某个大牛的向量思维对于解这种种类并查集特别有帮助,建议百度一下找找。
我的题解和大牛们的不一样啊首先说,我每个权值代表的方向基本和网上的题解都是相反的,比如 权值1 代表 x 吃 他的父亲节点,权值2 代表 x 被 他的父亲节点吃。其实这样的话就是相当于我建的树和他们的是相反的。
如果WA的话,可以单步运行一下Union函数看一看,有可能在路径压缩的时候没有成功更新权值。
欢迎提问~~~共同交流嘛。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 50010;
int fa[MAXN], relation[MAXN];
void Init(int n){
for(int i = 0;i <= n;++i){
fa[i] = i;
relation[i] = 0;
}
}
int Find(int x){
if(fa[x] == x)
return x;
else{
int temp = Find(fa[x]);
relation[x] = (relation[ fa[x] ] + relation[x] ) % 3;
fa[x] = temp; //这里无限WA……注意在更新父亲节点前更新权值。
return temp;
}
}
bool Union(int x, int y, int d){
int faX = Find(x), faY = Find(y);
if(faX != faY){
fa[faX] = faY; //这里要注意一下,谁是父亲节点会影响后续权值的更新
relation[faX] = ( -relation[x] + 3 + (d - 1) + relation[y] ) % 3;
return true; // 上面的这个表达式可以画几个图,然后自己试一试就明白了。
}
return false;
}
int main()
{
//freopen("input.txt", "r", stdin);
int n, k, ans = 0;
scanf("%d %d", &n, &k);
Init(n);
for(int i = 0;i < k;++i){
int d, x, y;
scanf("%d %d %d", &d, &x, &y);
if(x > n || y > n)
ans++;
else if(d == 2 && x == y) //自己吃自己
ans++;
else{ // 我储存relation数组时,0 代表 x 和 fa[x]是同类,1 代表 x 吃 fa[x], 2 代表 x 被 fa[x] 吃。因为这里吃和被吃是有明确向量关系的,所以要用三种数字表示。
if( !Union(x, y, d) ){ //如果Union失败,表示x 和 y父亲节点相同
int temp = (relation[x] - relation[y] + 3) % 3; // temp这里表示的是x 对 y的关系
if(temp != d - 1)
ans++;
}
}
}
printf("%d\n",ans);
return 0;
}