您的位置:首页 > 其它

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

2015-08-03 18:20 459 查看
题意:有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;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: