CF1408E Avoid Rainbow Cycles

一、题目

点此看题

本题的翻译似乎叫彩虹圈，好浪漫呀$qwq$

二、解法

给边染色然后找环特别不好做，而且边数都会炸。考虑把集合里面的所有点接在这个集合上，这个集合就可以代表颜色，用边的话可以通过到这个新建点中转完成，仔细考虑这样建出来是一个二分图。

考虑这个图的性质，如果这个图存在环，那么原图也存在环，因为可以用点到集合的边等效代换原图中点到点的边。问题变成了保留权值最大的边使得这个图不存在环，直接最大生成树。

#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 200005;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,tot,f[M],a[M],b[M],fa[M];long long ans;
struct node
{
	int u,v,c;
	bool operator < (const node &b) const
	{
		return c>b.c;
	}
}s[M];
int find(int x)
{
	if(x!=fa[x]) fa[x]=find(fa[x]);
	return fa[x];
}
signed main()
{
	m=read();n=read();
	for(int i=1;i<=m;i++)
		a[i]=read();
	for(int i=1;i<=n;i++)
		b[i]=read();
	for(int i=1;i<=m;i++)
	{
		int k=read();
		while(k--)
		{
			int j=read(),c=a[i]+b[j];
			s[++tot]=node{i+n,j,c};ans+=c;
		}
	}
	for(int i=1;i<=n+m;i++)
		fa[i]=i;
	sort(s+1,s+1+tot);
	for(int i=1;i<=tot;i++)
	{
		int u=s[i].u,v=s[i].v,c=s[i].c;
		int x=find(u),y=find(v);
		if(x==y) continue;
		fa[x]=y;ans-=c;
	}
	printf("%lld\n",ans);
}