LightOJ-1123-生成树

题目大意:给定图上一共有n的点,有m天,每天加一条边并且问当前的边集合的最小生成树是多少,没有的话就输出-1;

题目解析:每次都来个Kruskal肯定会T,其实到后来我们只需要保存n-1条边即可,因为其余边都是多余的,此时再加进来边只需找出那个能使得树变成环的边并且把它删除即可,因为每次肯定是删除一个,所以只需要令最后一个等于要删除的位置就ok了;

AC代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
struct edge
{
	int w,u,v;	
}e[6010];
int fa[301],n,m,cnt,c;
bool cmp(edge a,edge b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x==fa[x])	return x;
	else return	fa[x]=find(fa[x]);
}
void solve()
{
	sort(e,e+cnt,cmp);
	int i,ans=0,pos=-1,sum=0;
	for(i=0;i<cnt;i++)
	{
		int x=find(e[i].u);
		int y=find(e[i].v);
		if(x==y)
		{
			pos=i;
			continue;
		}
		else 
		{
			fa[x]=y;
			sum+=e[i].w;
			ans++;
		}
	}
	if(pos!=-1)
	{
		e[pos]=e[cnt-1];
		cnt--;
	}
	if(ans!=n-1)	
		printf("-1\n");
	else 
		printf("%d\n",sum);
}
int main()
{
	int cas,i,j;
	scanf("%d",&cas);
	for(c=1;c<=cas;c++)
	{
		scanf("%d%d",&n,&m);
		cnt=0;
		printf("Case %d:\n",c);
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&e[cnt].u,&e[cnt].v,&e[cnt].w);
			cnt++;
			for(j=1;j<=n;j++)
				fa[j]=j;
			solve(); 
		}
	}	
	return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值