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

题意:有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;
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值