码迷,mamicode.com
首页 > 其他好文 > 详细

!HDU 4313 破坏导弹攻城计划-连通图-(最小生成树变形)

时间:2015-08-03 19:08:55      阅读:181      评论:0      收藏:0      [点我收藏+]

标签:hdu   kruskal   

题意:有n个城市,城市之间有n-1条路连接每个城市,也就是说有一个树,敌人在一些城市放了导弹,共有m个导弹,敌人想把导弹运到一起然后开战,现在给你个任务就是摧毁一些路,使任何两个导弹都不能运到一起,摧毁每条路都有相应的代价,求最小的代价。

分析:

导弹不能运到一起就是说任何两个导弹不能在一个连通图中,所以我们的任务就是用最少的代价摧毁一些路使m个导弹分别在m个连通图中。

这题是就反面,最少的代价,那么就是剩下的路的代价越大越好,这就是最大生成树,所以解题方法是:用Kruskal算法求最大生成树,用并查集来维护两个导弹不能在一个连通图中。

代码:

#include<iostream>
#include<algorithm>
using namespace std;
int fa[100005],n,m,t;
struct node{
	int x,y;
	long long v;
}edge[100005];
int k[100005];
long long sum;
bool cmp(node a,node b)
{
	return a.v>b.v;
}
int find(int x)
{
	if(fa[x]==x) return x;
	else return fa[x]=find(fa[x]);
}
void merge(int x,int y)
{
	fa[find(x)]=find(y);
}
void kruskal()
{
	for(int i=0;i<n-1;i++){
		int fax=find(edge[i].x);
		int fay=find(edge[i].y);
		if(k[fax]&&k[fay]){
			sum+=edge[i].v;
			continue;
		}
		else{
			if(k[fax]||k[fay]) k[fax]=k[fay]=1;
			merge(fax,fay);
		}
	}
}
int main()
{
	cin>>t;
	while(t--){
		memset(k,0,sizeof(k));
		cin>>n>>m;
		for(int i=0;i<n;i++) fa[i]=i;
		for(int i=0;i<n-1;i++) cin>>edge[i].x>>edge[i].y>>edge[i].v;
		int tmp;
		for(int i=0;i<m;i++){
			cin>>tmp;
			k[tmp]=1;
		}
		sort(edge,edge+n-1,cmp);
		sum=0;
		kruskal();
		cout<<sum<<endl;
	}
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

!HDU 4313 破坏导弹攻城计划-连通图-(最小生成树变形)

标签:hdu   kruskal   

原文地址:http://blog.csdn.net/ac_0_summer/article/details/47260047

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!