Codeforces Round #543 D. Power Tree 最小生成树

题目链接： http://codeforces.com/contest/1120/problem/D

题意：

一棵结点带权的树，你要先选择其中若干个结点，同时付出他们的花费和，然后保证以这些结点为根的子树上的叶子加减上一定的值，可以让这些叶子在任意初始值的状态下变成 $0$ 。

做法：

因为要保证叶子的值无论在什么初始状态下都能最后变成 $0$ ，稍微推一下其实是可以感觉到最少选叶子结点的数量的值就可以。

一开始对题目的理解好像有点问题，感觉一个结点可以替代下面的值什么的，差点就用了左偏树什么的。。

然后就感觉着可以用树形 $dp$ ，但是由于个人能力有限…实在是没想到转移方程…最后默默的跟着学了一个新想法，最小生成树。

因为一个结点只能改变自己子树范围内的值，所以先把所有的叶子结点按照 $dfs$ 序组成一个新序列，假设某个结点 $i$ 可以影响到的范围是 $[l,r]$，那么可以把 $l$ 向 $r+1$ 连一条代价为 $val[i]$ 的边，表示其中的点都被覆盖了。因为边数比点数少 $1$ ，所以最后算的时候要把点数 $+1$ 。

最后最小生成树跑出来就好啦。至于答案要的东西，在中间各种怎么样做一下处理就好了。

努力努力！

代码

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define rep_e(i,u) for(int i=head[u];~i;i=nex[i])
using namespace std;
typedef long long ll;
const int maxn=200005;
const int maxm=400005;
const int inf=0x3f3f3f3f;
int va[maxn],n,m,fa[maxn],cou,lef[maxn],rig[maxn];
int head[maxn],to[maxm],nex[maxm],cnt,flag[maxn];
ll ans;
struct edge{
    int u,v,id;
    ll val;
    bool operator < (const edge &a)const{
        return val<a.val;
    }
}e[maxn];
void add(int u,int v){
    to[cnt]=v;nex[cnt]=head[u];
    head[u]=cnt++;
}

int fin(int x){
    return fa[x]==x?x:fa[x]=fin(fa[x]);
}
void dfs(int u,int f){
    int fl=0;
    lef[u]=inf,rig[u]=-1;
    rep_e(i,u){
        int v=to[i];
        if(v==f) continue;
        dfs(v,u); fl=1;
        lef[u]=min(lef[u],lef[v]);
        rig[u]=max(rig[u],rig[v]);
    }
    if(!fl){
        lef[u]=rig[u]=++cou;
    }
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    rep(i,1,n) scanf("%d",&va[i]);
    rep(i,2,n){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    dfs(1,-1);
    rep(i,1,n) e[i]=(edge){lef[i],rig[i]+1,i,va[i]};
    sort(e+1,e+1+n);
    rep(i,1,cou+1) fa[i]=i;
    int nump=0;
    rep(i,1,n){
        int nex=i;
        while(nex+1<=n&&e[nex+1].val==e[i].val) nex++;
        rep(k,i,nex){
            int u=e[k].u,v=e[k].v;
            u=fin(u),v=fin(v);
            if(u!=v) flag[e[k].id]=1,nump++;
        }
        rep(k,i,nex){
            int u=e[k].u,v=e[k].v;
            u=fin(u),v=fin(v);
            if(u!=v) fa[u]=v,ans+=e[k].val;
        }
        i=nex;
    }
    printf("%lld %d\n",ans,nump);
    rep(i,1,n){
        if(flag[i]){
            printf("%d%c",i,(--nump==0?'\n':' '));
        }
    }
    return 0;
}