Uva 10807 - Prim

看着像是一个有重边的图生成2个无重边的总权值最小的生成树

实际上却是一道暴力搜索的题目

搜索不TLE的关键点在于两个生成树一定有的差别是一个有最短边,一个没有

所以枚举第一个生成树的时候一定要有最小边即可

写的姿势比较差

TLE无数,还是看了discuss才知道的这点=.=

9Y

最开始的想法是两遍贪心,果断WA

附测试用例于代码后

#include<stdio.h>
#include<vector>
#include<algorithm>
#define MAX 30
struct edge{
	int p,q,w;
	bool operator < (const edge& b)const {
		return w<b.w;
		}
	};
std::vector<edge>L;
int ft[MAX];
void in_ft(int n){
	int i;
	for(i=1;i<=n;i++)ft[i]=i;
	}
int find(int x){
	int t=x,temp;
	while(ft[x]!=x)x=ft[x];
	while(t!=x){
		temp=ft[t];
		ft[t]=x;
		t=temp;
		}
	return x;
	}
void merge(int root1,int root2){
	int fr1=find(root1);
	int fr2=find(root2);
	ft[fr1]=fr2;
	}
int res=(int)1e9;
int cho[MAX];
bool ok[MAX];
int n,m;
int min(int a,int b){
	if(a>b)return b;return a;
	}
int max(int a,int b){
	if(a<b)return b;return a;
	}
int is_tree(){
	int i;
	in_ft(n);
	for(i=1;i<n;i++){
		int t=cho[i];
		int p=L[t].p;
		int q=L[t].q;
		if(find(p)!=find(q))merge(p,q);
		else return 0;
		}
	for(i=2;i<=n;i++)if(find(i)!=find(1))return 0;
	return 1;
	}
void DFS(int t){
	int i;
	if(t==n){
		int flag=is_tree();
		if(!flag)return;
	//	for(i=1;i<n;i++)printf("%d ",cho[i]);printf("\n");
		for(i=0;i<m;i++)ok[i]=0;
		for(i=1;i<n;i++)ok[ cho[i] ]=1;
		in_ft(n);
		int ha=0;
		for(i=cho[1]+1;i<m;i++){
			if(ok[i])continue;
			int p,q;
			p=L[i].p;
			q=L[i].q;
			if(find(p)!=find(q))merge(p,q),ha+=L[i].w;
			}
		find(1);
		for(i=2;i<=n;i++)if(find(i)!=ft[1])return;
		for(i=1;i<n;i++)ha+=L[ cho[i] ].w;
		res=min(res,ha);
		}
	else {
		i=max(cho[t-1]+1,t-1);
		for(;i<=m-n+t-2;i++){
			cho[t]=i;
			DFS(t+1);
			}
		}
	}
int main(){
	while(scanf("%d",&n)!=EOF&&n){
		in_ft(n);L.clear();res=(int)1e9;
		scanf("%d",&m);
		int i,p,q,w;
		for(i=1;i<=m;i++){
			scanf("%d%d%d",&p,&q,&w);
			L.push_back( (edge){p,q,w} );
			}
		std::sort(L.begin(),L.end());
		cho[1]=0;
		DFS(2);
		if(res==(int)1e9)printf("No way!\n");
		else printf("%d\n",res);
		}
	return 0;
	}

4
7
1 2 1
1 2 1
2 3 1
2 4 10
2 4 20
3 4 1
3 4 100
0
AC out:34

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值