首先说明这是并查集的一个经典应用,真的是一道非常不错的题目。
以上是题目的描述,首先这是一个并查集的题目所以要先熟练掌握并查集(可以参考以下链接和题目):
https://blog.csdn.net/weixin_44758733/article/details/104154271
POJ 2524 添加链接描述
POJ 1611 添加链接描述
#总体思路:
如果两个结点确定了关系,则将这两个结点添加到同一棵树中。
在并查集的基础上增加一个rank[]数组以记录每个结点和根结点
那么在同一棵树中会有以下3种关系:
0: 和父亲节点同类
1: 被父亲节点吃
2: 吃父亲节点
首先对输入的数据是否合法进行判断,不合法直接谎言书+1。
进一步判断输入的两个结点:
首先的前提是在同一棵数中x和y无非就是动物A,B,C中的一种,可以的到以下的图:
再来解释在食物链中的一个重要的关系就是:
x -> y, y -> z如果已知那么就可以得到x -> z的关系(枚举法证明):
最后一个重要的问题就是如果两个结点不在同一棵树中的情况即两棵树的合并
我采用的策略是固定的将y树合并到x树上边去(也可以将x树合并到y书上边,代码类似),其实合并并不难主要是对rank[]数组的更新(fx,fy分别是相应树的根节点)。
rank[y]存储的是y -> fy的关系 那么fy -> y的关系就是其逆关系即(3-rank[y])%3
至于合并的x和y的关系:
当d=1时,y和x同类 y->x = 0;
当d=2时,y被x吃 y->x = 1;
综上y->x = d-1;
x->fx 就是rank[x]
则综上可知 fy -> fx = (3-rank[y] + d-1 + rank[x])%3
以上所有的都已分析完毕,以下是代码:
首先说明以下这道题,如果写多组输入本题会WA,希望知道的大佬可以将答案留在评论区,Thanks~!
#include <cstdio>
using namespace std;
const int animals_num = 5e4 + 5;
int father[animals_num],Rank[animals_num];
void initialize(int n)
{
for (int i=1;i<=n;i++)
{
father[i] = i;
Rank[i] = 0;
}
}
int find(int son)
{
if (son==father[son])
return son;
int root = father[son];
father[son] = find(father[son]);
Rank[son] = (Rank[son] + Rank[root])%3;
return father[son];
}
void Union(int x,int y,int d)
{
int fx = find(x);
int fy = find(y);
father[fy] = fx;
Rank[fy] = (3-Rank[y] + d-1 + Rank[x])%3;
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
initialize(n);
int type,x,y;
int ans = 0;
while (k--)
{
scanf("%d%d%d",&type,&x,&y);
if (x>n || y>n) ans++;
else if (2==type && x==y) ans++;
else
{
if (find(x) == find(y))
{
if (1==type && Rank[x] != Rank[y]) ans++;
if (2==type && (Rank[x]+1)%3 != Rank[y]) ans++;
}
else
Union(x,y,type);
}
}
printf("%d\n",ans);
return 0;
}