tyvj P3737 逐个击破

Posted FuTaimeng

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tyvj P3737 逐个击破相关的知识,希望对你有一定的参考价值。

http://tyvj.cn/p/3737

时间: 1000ms / 空间: 131072KiB / Java类名: Main

描述

秉承伟大军事家的战略思想,作为一个有智慧的军长你,遇到了一个类似的战场局面:

现在有N个城市,其中K个被敌方军团占领了,N个城市间有N-1条公路相连,破坏其中某条公路的代价是已知的,现在,告诉你K个敌方军团所在的城市,以及所有公路破坏的代价,请你算出花费最少的代价将这K个地方军团互相隔离开,以便第二步逐个击破敌人。

输入格式

第一行包含两个正整数n和k。

第二行包含k个整数,表示哪个城市别敌军占领。

接下来n-1行,每行包含三个正整数a,b,c,表示从a城市到b城市有一条公路,以及破坏的代价c。

城市的编号从0开始计数。

其中:

2<=n<=100000

2<=k<=n

1<=c<=1000000

输出格式

包含一个整数,表示最少花费的代价。

测试样例1

输入

3 3

0 1 2

0 1 1

1 2 2

输出

3

测试样例2

输入

5 3

1 2 4 

1 0 4 

1 3 8 

2 1 1

2 4 3

输出

4

破坏的最少=留下的最多,

使最多一个敌军驻地包含在内一个连通子图内

做最大生成树,需要合并连个同在敌占区的组合时,这条边就是应该被破坏的。

例如:7-9不需要破坏,将9并入敌占区;

           6-7不需要破坏,将6并入敌占区;

          6,8同在敌占区, 6-8 需要破坏,这是连接7,8的最短路。

同理,将5并入敌占区

将1,2,4并入3对应的敌占区;

最后,将4-5切断。

 

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define Size 100005

int n,k;
struct edge{
    int u,v,w;
}eg[Size];
long long ans=0;
bool tag[Size];

int pa[Size];
void init(){
    for(int i=1;i<=n;i++)pa[i]=i;
}
int find(int x){
    if(pa[x]!=x)pa[x]=find(pa[x]);
    return pa[x];
}

bool ff(edge a,edge b){
    return a.w>b.w;
}

void kruskal(){
    init();
    sort(eg+1,eg+n,ff);
    
    for(int i=1;i<=n-1;i++){
        int x=find(eg[i].u);
        int y=find(eg[i].v);
        if(tag[x]&&tag[y]){
            ans+=eg[i].w;
        }
        else{
            tag[y]=tag[x]||tag[y];
            pa[x]=y;
        }
    }
}

int main(){
    scanf("%d%d",&n,&k);
    int x;
    for(int i=1;i<=k;i++){
        scanf("%d",&x);
        tag[x]=true;
    }
    for(int i=1;i<=n-1;i++){
        scanf("%d%d%d",&eg[i].u,&eg[i].v,&eg[i].w);
    }
    kruskal();
    printf("%lld",ans);
}

注意一定用scanf

用cin超时。。。

 

以上是关于tyvj P3737 逐个击破的主要内容,如果未能解决你的问题,请参考以下文章

P2700 逐个击破

洛谷P2700 逐个击破

p2700 逐个击破

luoguP2700 逐个击破

题解逐个击破 luogu2700

题解逐个击破 luogu2700