P2700 逐个击破

  一开始以为题很难,当我坐下来认真读题后,却神奇地发现这道题很水……

  这和Kruskal有什么区别啊?

  于是我开开心心地十五分种敲完了代码~

  然后我就Wa了……

  emmm……

  尴尬……

  一开始我对于每一个点的no[]值都进行维护,时间复杂度特别高,于是就Wa了qwq

  后来我发现,其实一颗树只要维护一个no,即根结点的no即可,这样判断这棵树数是否包含那k个城市的时候。只要看根节点就可以了。

  那么对于树的相连,看哪一棵树的根节点有no,就继续作为根节点;如果都没有,直接加边就可以,如果都有,这条边就不能连啦~

  简单地总结一下,先贪心排序,再正常连边,判断是否能连,一访问到no就直接将其作为根结点,便于以后的访问。

  还有,开long long(别问我是怎么知道的)……

  其实写代码的时候注意到有关no转移的技巧,这就是一道提高减的题,根本没什么难度。

  上代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 100005
#define int long long
struct data
{
    int x,y,w;
} q[maxn];
int n,k,sum,par[maxn];
bool no[maxn],flag;
bool cmp(data a,data b)
{
    return a.w>b.w;
}
int find(int x)
{
    if(x==par[x])
        return x;
    return par[x]=find(par[x]);
}
main()
{
    scanf("%lld%lld",&n,&k);
    for(int i=0; i<n; i++)
        par[i]=i;
    for(int i=0; i<k; i++)
    {
        int a;
        scanf("%lld",&a);
        no[a]=1;
    }
    for(int i=0; i<n-1; i++)
    {
        int a,b,c;
        scanf("%lld%lld%lld",&a,&b,&c);
        q[i].x=a;
        q[i].y=b;
        q[i].w=c;
    }
    sort(q,q+n-1,cmp);
    for(int i=0; i<n-1; i++)
    {
        if(no[find(q[i].x)]&&no[find(q[i].y)])
        {
            sum+=q[i].w;
            continue;
        }
        int nx=find(q[i].x);
        int ny=find(q[i].y);
        if(no[ny])
            par[nx]=ny;
        else
            par[ny]=nx;
    }
    printf("%lld",sum);
    return 0;
}

 

posted @ 2018-12-02 17:35  paopo  阅读(222)  评论(0编辑  收藏  举报