CF1444C Team-Building

一、题目

点此看题

二、解法

要判断二分图可以使用带权并查集，并查集上的边设置为 $1$ 就表示两点必须异色，设置为 $0$ 就保证两点必须同色，所以我们在相同连通块查询合法时路径上权值异或和必须是 $1$

补充一点，上面的问题其实是判断一棵树上加入若干条边是否形成奇环，这个问题是出过题的。只用考虑树边和一条非树边形成的环是可以证明的（这里就不证明了，分类讨论即可）

我们用带权并查集首先维护颜色内部的边，首先对于自己就不行的颜色排除在外，设剩下的颜色数量为$s$，那么答案可以表示成：$\frac{s(s-1)}{2}-shit$，其中 $shit$ 表示不合法的方案。

现在考虑如何求出不合法方案，发现如果两个颜色没有边相连那么一定合法。我们只考虑有边相连的颜色，由于边的数量时有限的所以这个算法可行。

把同类边（连接颜色相同的边）放在一起，每次把他们都取出来然后暴力连接判断，就用带权并查集的方法。但是需要支持回退，不能用路径压缩但又要保证复杂度。那么就用启发式合并呐，可以保证树形态的相对固定，并且时间复杂度 $O(n\log n)$

#include <cstdio>
#include <vector>
#include <iostream>
#include <map>
using namespace std;
const int M = 500005;
#define pii pair<int,int>
#define make make_pair
#define ll long long
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,k,cnt,a[M],b[M],c[M],w[M],fa[M],siz[M],jzm[M];
vector<pii> v[M];map<ll,int> mp;ll ans;vector<int> g;
int ask(int x)
{
	if(x==fa[x]) return 0;
	return c[x]^ask(fa[x]);
}
int find(int x)
{
	if(x==fa[x]) return x;
	return find(fa[x]);
}
int merge(int u,int v,int fl)
{
	int x=find(u),y=find(v),z=ask(u)^ask(v)^1;
	if(x==y)
	{
		//printf("->%d\n",z^1);
		return z^1;
	}
	if(siz[x]>siz[y])
	{
		if(fl) g.push_back(y);
		fa[y]=x;c[y]=z;
		siz[x]+=siz[y];
	}
	else
	{
		if(fl) g.push_back(x);
		fa[x]=y;c[x]=z;
		siz[y]+=siz[x];
	}
	return 1;
}
void clear()
{
	while(g.size())
	{
		int x=g.back();g.pop_back();
		siz[fa[x]]-=siz[x];
		fa[x]=x;c[x]=0;
	}
}
signed main()
{
	n=read();m=read();k=read();
	for(int i=1;i<=n;i++)
	{
		w[i]=read();
		siz[i]=1;fa[i]=i;
	}
	for(int i=1;i<=m;i++)
		a[i]=read(),b[i]=read();
	for(int i=1;i<=m;i++)
		if(w[a[i]]==w[b[i]])
		{
			if(!merge(a[i],b[i],0))
				jzm[w[a[i]]]=1;
		}
	for(int i=1;i<=k;i++)
		if(!jzm[i]) ans++;
	ans=ans*(ans-1)/2;
	//printf("%lld\n",ans);
	for(int i=1;i<=m;i++)
	{
		if(jzm[w[a[i]]]) continue;
		if(jzm[w[b[i]]]) continue;
		if(w[a[i]]!=w[b[i]])
		{
			if(w[a[i]]>w[b[i]]) swap(a[i],b[i]);
			ll d=1ll*w[a[i]]*k+w[b[i]];
			if(!mp[d]) mp[d]=++cnt; 
			v[mp[d]].push_back(make(a[i],b[i]));
		}
	}
	for(int i=1;i<=cnt;i++)
	{
		//puts("---------"); 
		for(int j=0;j<v[i].size();j++)
		{
			int x=v[i][j].first,y=v[i][j].second;
			//printf("%d %d\n",x,y);
			if(!merge(x,y,1))
			{
				ans--;
				break;
			}
		}
		clear();
	}
	printf("%lld\n",ans);
}