洛谷 P2024 食物链

P2024 食物链

法一：种类并查集

#include<cstdio>
int fa[250000];
int n,k,ans;
//fa[i]表示与i同类的集合，fa[i+n]表示i吃的集合，fa[i+2n]表示吃i的集合
int find(int x)
{
	if(fa[x]!=x)	fa[x]=find(fa[x]);
	return fa[x];
}
//int union1(int x,int y)
//{
//	x=find(x);
//	y=find(y);
//	fa[x]=y;
//}//直接写进主程序 
int main()
{
	int i,a,b,c,f1,f2,f3,f4;
	scanf("%d%d",&n,&k);
	for(i=1;i<=3*n;i++)
		fa[i]=i;
	for(i=1;i<=k;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		if(b>n||c>n||(a==2&&b==c))
			ans++;
		else
		{
			if(a==1)
			{
				f1=find(b);
				f2=find(c);
				f3=find(c+n);
				f4=find(c+2*n);
				if(f1==f3||f1==f4)//如果已知b被c吃或b吃c则为假话
					ans++;
				else
					if(f1!=f2)
					{
						fa[f1]=f2;//与b同类的集合和与c同类的集合合并
						fa[find(b+n)]=f3;//b吃的集合和c吃的集合合并
						fa[find(b+2*n)]=f4;//吃b的集合和吃c的集合合并
					}
			}
			if(a==2)
			{
				f1=find(b);
				f2=find(c);
				f3=find(c+n);
				f4=find(c+2*n);
				if(f1==f2||f1==f3)//如果已知b和c同类或b被c吃则为假话
					ans++;
				else
					if(f1!=f4)
					{
						fa[find(b+n)]=f2;//b吃的集合和与c同类的集合合并
						fa[find(c+2*n)]=f1;//吃c的集合和与b同类的集合合并
						fa[find(b+2*n)]=f3;//吃b的集合和c吃的集合合并 
						//！！！上面这种非常容易忘，当心了 
					} 
			}
		}
	}
	printf("%d",ans); 
	return 0;
}

法二：加权并查集

加权并查集的做法并非将同类放入同一个并查集，而是将所有有关系的动物都放入同一个并查集。这一点跟上面的做法不同。
r[i]表示i结点与父亲结点的关系，r[i]=0 表示father[i]与i同类；1表示father[i]吃i；2表示i吃father[i]。
在这道题中，处理合并与路径压缩时的权值变化的计算方法基本上都由枚举、找规律得出。事实上，即使没有规律，或许也可以通过直接写大量if判断语句来处理权值变化。处理权值变化的方法需要具体情况具体分析。

#include<cstdio>
#include<algorithm>
using namespace std;
int father[60000];
int r[60000];
int n,k;
int find(int x)
{
    if(x!=father[x])
    {
        int xx=father[x];
        father[x]=find(father[x]);
        r[x]=(r[x]+r[xx])%3;//见“解释4”
    }
    return father[x];
}
int ans;
int main()
{
    int a,i,b,c,fb,fc;
    scanf("%d%d",&n,&k);
    for(i=1;i<=n;i++)
        father[i]=i;
    for(i=1;i<=k;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        if(b>n||c>n||(b==c&&a==2))//第二和第三条判定假话的方法
            ans++;
        else
        {
            fb=find(b);
            fc=find(c);
            if(a==1)
            {
                if(fb==fc)//b和c在相同集合中，即能确定b和c有关系
                {
                    if(r[b]!=r[c])//在find中已经完成路径压缩，因此r[b]、r[c]表示它们与同一个结点的关系
                        ans++;//如果它们与该结点关系相同，显然b、c同类，不矛盾，反之则矛盾，答案加1
                }
                else//如果不能确定b、c有关系
                {
                    father[fc]=fb;//合并b、c所在集合
                    r[fc]=(r[b]-r[c]+3)%3;//这个不好解释，见下方“解释1”
                }
            }
            else//如果输入是b吃c
            {
                if(fb==fc)//如果可以确定b与c的关系
                {
                    if((r[b]+1)%3!=r[c])//如果b不是吃c的（原因见“解释2”）
                        ans++;//答案加1
                }
                else
                {
                    father[fc]=fb;//如果不能确定b与c关系，则合并b和c所在集合
                    r[fc]=(r[b]-r[c]+4)%3;//见“解释3”
                }
            }
        }
    }
    printf("%d",ans);
    return 0;
}

解释1：
可以知道此时b和c是同类，并且b和c所在的集合都已经完成了路径压缩，也就是b和c此时的父结点分别就是fb和fc。直接枚举r[b]、r[c]值，推出对应r[fc]值，找规律即可。
举例：r[b]=1,r[c]=2时，fb吃b，c吃fc，b和c同类，可得fc吃fb，即r[fc]=2。
很容易发现r[fc]=(r[b]-r[c]+3)%3。
解释2：
b和c所在的集合都已经完成了路径压缩，且fb==fc，也就是b和c此时的父结点为同一个即fb。枚举r[b]值推出r[c]可找出规律。
举例：r[b]=1，那么b被fb吃，而如果b吃c，显然c吃fb，因此r[c]=2。
很容易发现当且仅当(r[b]+1)%3==r[c]时b是吃c的。
解释3：
可以知道此时b吃c，并且b和c所在的集合都已经完成了路径压缩，也就是b和c此时的父结点分别就是fb和fc。仍然是枚举r[b]、r[c]值推倒r[fc]值找规律。
举例：r[b]=1，r[c]=0，那么b被fb吃，而b吃c，因此c吃fb，而c与fc同类，因此fc吃fb，因此r[fc]=2。
很容易发现r[fc]=(r[b]-r[c]+4)%3。
解释4：
如果某点与其父结点关系为a，其父结点与根结点关系为b，那么可以知道该点与根结点关系为(a+b)%3。
举例：某点被父结点吃，即r[点]=1，父结点吃根结点，即r[father[点]]=2，那么显然该点与根结点同种类，即路径压缩后r[点]=(1+2)%3=0。
（这个不好证也没必要证，多列几组试一下就知道了）

推荐优秀讲解：

食物链
食物链