并查集——扩展域并查集和带权并查集

相关例题
食物链
D. The Number of Imposters

食物链

题目大意

给出一种食物链，其中A吃B，B吃C，C吃A，构成一种环形食物链
给出m个输入，其中包括两种输入k,x,y：
		1、当k = 1时，x 和 y为同类
		2、当k = 2时，表示 x 吃 y
这 m 个输入中有的是假话，当满足以下三个条件之一，即为假话
		1、当前的话和以前的话存在冲突
		2、x 或 y 的值比n大
		3、x 吃 x
求这m个输入中有多少假话

方法一（带权并查集）：

通过一个数组d来表示所有点到根节点的距离，默认为0，其中距离 %3 的三种结果0 ， 1， 2表示三种动物

过题代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 5e4 + 10;

int d[N],p[N];

int find(int x)
{
    if(x != p[x])
    {
        int t = find(p[x]);
        d[x] += d[p[x]];
        p[x] = t;
    }
    return p[x];
}

int main()
{
    int n,m;
    cin >> n >> m;
    
    for(int i = 1;i <= n;i ++)
        p[i] = i;
        
    int ans = 0;
    
    while (m -- )
    {
        int k,a,b;
        cin >> k >> a >> b;
        
        if(a > n || b > n)
        {
            ans ++ ;
            continue;
        }
        
        int fa = find(a),fb = find(b);
        
        if(k == 1)
        {
            if(fa == fb && (d[a] - d[b]) % 3)ans ++;//a 和 b不符合关系
            else if(fa != fb){
                p[fa] = fb;
                d[fa] = d[b] - d[a];
            }
        }else {
            if(fa == fb && (d[a] - d[b] - 1) % 3)ans ++;//a 和 b不符合关系
            else if(fa != fb)
            {
                p[fa] = fb;
                d[fa] = d[b] - d[a] + 1;
            }
        }
    }
    
    cout << ans << "\n";
    return 0;
}

方法二（扩展域并查集）

创造3*n的数组空间，x,x + n,x + n + n分别表示x的同类、捕食域和天敌域
当k = 1时，p[x + n] 和 p[x + n + n] 中含有y为假话
当k = 2时，p[x] 和 p[x +n +n]中含有y为假话

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 5e4 + 10;

int p[3 * N];

int n,m,k,a,b;

int find(int x)
{
    if(x != p[x])
        p[x] = find(p[x]);
    return p[x];
}

void me(int a,int b)
{
    int fa = find(a),fb = find(b);
    p[fa] = p[fb];
    return;
}

int main()
{
    cin >> n >> m;
    
    int ans = 0;
    
    for(int i = 1;i <= 3 * n;i ++)
        p[i] = i;
    
    while (m -- )
    {
        cin >> k >> a >> b;
        
        if(a > n || b > n){
            ans ++;
            continue;
        }
        
        if(k == 1)
        {
            if(find(a + n) == find(b) || find(a + n + n) == find(b))ans ++;//a的捕食域或天敌是b
            else {
                me(a,b);
                me(a + n,b + n);
                me(a + n + n,b + n + n);
            } 
        }else {
            if(find(a) == find(b) || find(a + n + n) == find(b))ans ++;//a的同类或者天敌是b
            else {
                me(a,b + n + n);
                me(a + n,b);
                me(a + n + n, b + n);
            }
        }
    }
    
    cout << ans << "\n";
    return 0;
}

D. The Number of Imposters（cf 1594D）

题目大意

与食物链题类似，对于x,y,ch 共有两种输入方式
		1、当ch = imposter 时，表示 x 说 y 是坏人
		2、当ch = crewmate时，表示 x 说 y 是好人
并且好人说的都是真话，坏人说的都是假话

题目分析

通过分析容易发现，当 x 说 y 是好人时，不论 x 是不是好人，x 和 y 都是一伙的；当 x 说 y 是坏人时，不论 x 是不是好人，x 和 y都是对立的
于是，我们可以用上面两种方法解决这道问题

方法一（带权并查集）

解题代码

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=2e18+100;
const int maxn=2e5+100;
int a[maxn],pre[maxn],cnt[maxn][2];
int n,m;
int find(int x)
{
	if(x!=pre[x])
	{
		int root=find(pre[x]);
		a[x]^=a[pre[x]];
		pre[x]=root;
	}
	return pre[x];
}
signed main()
{
	IOS
	int tt;
	cin>>tt;
	while(tt--)
	{
		bool fg=1;
		cin>>n>>m;
		for(int i=1; i<=n; i++)
		{
			pre[i]=i;
			a[i]=0;
			cnt[i][0]=1;
			cnt[i][1]=0;
		}
		for(int i=1; i<=m; i++)
		{
			int x,y;
			string str;
			cin>>x>>y>>str;
			int val=0;
			if(str[0]=='i')val=1;
			int fx=find(x),fy=find(y);
			if(fx==fy)
			{
				if((a[x]^a[y])!=val)fg=0;
			}
			else
			{
				pre[fy]=fx;
				if((a[x]^a[y])!=val)
				{
					a[fy]^=1;
					cnt[fx][1]+=cnt[fy][0];
					cnt[fx][0]+=cnt[fy][1];
				}
				else
				{
					cnt[fx][1]+=cnt[fy][1];
					cnt[fx][0]+=cnt[fy][0];
				}
			}
		}
		int ans=0;
		for(int i=1;i<=n;i++)
		{
			if(pre[i]==i)ans+=max(cnt[i][0],cnt[i][1]);
		}
		if(fg==0)cout<<"-1"<<"\n";
		else cout<<ans<<"\n";
	}
}

方法二（扩展域并查集）

解题代码

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <cmath>
#include <cstring>
#include <set>
#include <list>
#include <unordered_set>
#include <unordered_map>

#define gcd __gcd
#define LL long long
#define PII pair<int,int>
#define MII map<int,int>
#define MIS map<int,string>
#define MSI map<string,int>
#define VI vector<int>
#define pb push_back
#define fi first
#define se second
#define all(x) (x).begin(),(x).end()

using namespace std;
const int N = 2e5 + 10,M = 1e5 + 10;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;

int p[2 * N],res[2 * N];
int n,m,a,b;
string s;

int find(int x)
{
    if(x != p[x])p[x] = find(p[x]);
    return p[x];
}

void me(int a,int b)
{
    int fa = find(a),fb = find(b);
    if(fa != fb)
    {
        p[fa] = fb;
        res[fb] += res[fa];
        res[fa] = 0;
    }
}

int main()
{
    int tt;
    cin >> tt;
    while(tt--)
    {
        cin >> n >> m;
        
        bool flag = true;
        for(int i = 1;i <= n;i ++)
            p[i] = i,res[i] = 1,res[i + n] = 0,p[i + n] = i + n;
            
        while(m--)
        {
            cin >> a >> b >> s;
            
            if(s[0] == 'i')
            {
                if((find(a) == find(b) || find(a + n) == find(b + n)))flag = false;
                else {
                    me(a,b + n);
                    me(a + n,b);
                }
            }else {
                if((find(a) == find(b + n)|| find(a + n) == find(b)))flag = false;
                else {
                    me(a , b);
                    me(a + n,b + n);
                }
            }
        }
        
        if(flag){
            int ans = 0;
            for(int i = 1;i <= n;i ++)
                ans += max(res[i],res[i + n]);
            cout << ans << "\n";
        }else cout << "-1\n";
    }
    return 0;
}