bzoj3252 攻略

http://www.elijahqi.win/2018/03/15/bzoj3252/ ‎
Description
题目简述：树版[k取方格数]
众所周知，桂木桂马是攻略之神，开启攻略之神模式后，他可以同时攻略k部游戏。今天他得到了一款新游戏《XX
半岛》，这款游戏有n个场景(scene)，某些场景可以通过不同的选择支到达其他场景。所有场景和选择支构成树状
结构：开始游戏时在根节点（共通线），叶子节点为结局。每个场景有一个价值，现在桂马开启攻略之神模式，同
时攻略k次该游戏，问他观赏到的场景的价值和最大是多少（同一场景观看多次是不能重复得到价值的）
“为什么你还没玩就知道每个场景的价值呢？”
“我已经看到结局了。”
Input
第一行两个正整数n,k
第二行n个正整数，表示每个场景的价值
以下n-1行,每行2个整数a,b，表示a场景有个选择支通向b场景（即a是b的父亲）
保证场景1为根节点
n<=200000，1<=场景价值<=2^31-1
Output
输出一个整数表示答案
Sample Input
5 2
4 3 2 1 1
1 2
1 5
2 3
2 4
Sample Output
10
贪心+树形dp即可 据说是scoi2017的第一题 但是因为据说锅很大成了nsoi就没有题放出..
考虑我现在要选k-1个 那么我是不是假设已经所有大的都选完了 那么我选第k个的时候显然只能在剩下中挑选 所以不妨遵照这个思想我第一遍dp求出每个节点最重的儿子是谁 然后第二遍就把我的只推给最重的那个儿子一直推到叶子节点 这样可以保障我一定使得最大的链的答案最大 所以我就直接选前k大即可

#include<vector>
#include<cstdio>
#include<algorithm>
#define ll long long
#define rg register
#define N 220000
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=gc();}
    while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=gc();
    return x*f;
}
ll dp[N],dp1[N];vector<ll> ans;int n,k,a[N],h[N],son[N],num;
struct node{
    int y,next;
}data[N<<1];
inline void dfs(int x,int fa){
    dp[x]=a[x];
    for (rg int i=h[x];i;i=data[i].next){
        int y=data[i].y;if (y==fa) continue;
        dfs(y,x);if (dp[y]>dp[son[x]]) son[x]=y;
    }dp[x]+=dp[son[x]];
}
inline void dfs1(int x,int fa){
    dp1[x]+=a[x];bool flag=0;
    for (rg int i=h[x];i;i=data[i].next){
        int y=data[i].y;if (y==fa) continue;flag=1;
        if (y==son[x]) dp1[y]+=dp1[x];dfs1(y,x);
    }if (!flag) ans.push_back(dp1[x]);
}
inline bool cmp(const ll &a,const ll &b){
    return a>b;
}
int main(){
    freopen("bzoj3252.in","r",stdin);
    n=read();k=read();
    for (rg int i=1;i<=n;++i) a[i]=read();
    for (rg int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].y=y;data[num].next=h[x];h[x]=num;
        data[++num].y=x;data[num].next=h[y];h[y]=num;
    }dfs(1,1);dfs1(1,1);
    sort(ans.begin(),ans.end(),cmp);ll sum=0;
    for (rg int i=0;i<k;++i) sum+=ans[i];printf("%lld\n",sum);
    return 0;
}