CF875F Royal Questions

一、题目

点此看题

二、解法

把王子当成点，公主当成边，边有边权，每个点只能选一个相连的边，要求最大化边权。

类似于生成树问题，只不过我们允许基环树，也就是在生成树的基础上多一条边。可以用最小生成树的贪心，把边权从大到小排序，两个点不同根并且不全是基环树就合并，如果相同根看是不是基环树，不是的话添加这条边。

#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 200005;
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,ans,fa[M],s[M];
struct node
{
	int u,v,c;
	bool operator < (const node &b) const
	{
		return c>b.c;
	}
}a[M];
int find(int x)
{
	if(fa[x]!=x) fa[x]=find(fa[x]);
	return fa[x];
}
signed main()
{
	n=read();m=read();
	for(int i=1;i<=m;i++)
	{
		a[i].u=read();a[i].v=read();a[i].c=read();
	}
	sort(a+1,a+1+m);
	for(int i=1;i<=n;i++)
		fa[i]=i,s[i]=1;
	for(int i=1;i<=m;i++)
	{
		int x=find(a[i].u),y=find(a[i].v);
		if(x!=y && (s[x] || s[y])) fa[x]=y,s[y]&=s[x],ans+=a[i].c;
		else if(s[x]) s[x]=0,ans+=a[i].c;
	}
	printf("%d\n",ans);
}