【BZOJ】【P3624】【Apio2008】【免费道路】【题解】【生成树】

最新推荐文章于 2021-05-17 20:22:33 发布

原创最新推荐文章于 2021-05-17 20:22:33 发布 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#bzoj

OI 专栏收录该内容

324 篇文章

订阅专栏

本文介绍了一种基于最小生成树的算法实现方案，通过两遍处理来解决特定问题：首先生成一棵包含特定数量零权边的树，若零权边数量过多则返回无解；其次补充剩余的零权边并使用单位权边完成树的构建。

传送门:http://www.lydsy.com:808/JudgeOnline/problem.php?id=3624

先做一遍生成树,有多于k个0边就no solution

再把原来选的0边加上,再选一些0边凑够k个

再用1边补满

完了

Code:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
struct edge{
	int u,v;
}edges[maxn];
int fa[maxn];
int n,m,k,cnt;
int front,back;
bool vis[maxn];
int find(int x){
	if(fa[x]!=x)return fa[x]=find(fa[x]);return x;
}
int main(){
	scanf("%d%d%d",&n,&m,&k);back=m;
	for(int i=1;i<=m;i++){
		int u,v,c;scanf("%d%d%d",&u,&v,&c);
		if(!c)edges[back--]=(edge){u,v};
		else edges[++front]=(edge){u,v};
	}for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++){
		if(find(edges[i].u)!=find(edges[i].v)){
			fa[find(edges[i].u)]=find(edges[i].v);
			vis[i]=i>front;	cnt+=i>front;
		}
	}if(cnt>k){
		puts("no solution");
		return 0;
	}for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=back+1;i<=m;i++)
		if(vis[i])fa[find(edges[i].u)]=find(edges[i].v);	
	for(int i=back+1;i<=m;i++)
		if(!vis[i]&&cnt<k&&find(edges[i].u)!=find(edges[i].v))
		fa[find(edges[i].u)]=find(edges[i].v),cnt++,vis[i]=1;
	if(cnt!=k){
		puts("no solution");
		return 0;		
	}
	for(int i=1;i<=front;i++)
		if(find(edges[i].u)!=find(edges[i].v))
		fa[find(edges[i].u)]=find(edges[i].v),vis[i]=1;
	for(int i=1;i<=m;i++)if(vis[i])
	printf("%d %d %d\n",edges[i].u,edges[i].v,i<=front);		
	return 0;
}