HDU 5242 Game(树上贪心)

题目链接 Game

题目的意思很简单， 就是要找一棵树权值最大等等前K条链。

在本题中，走的次数等于min(叶子结点个数，k)

tree[i].sum意为从i号结点出发走到某个叶子结点能得到的最大总价值。

pson[i]表示i号结点若要获得最大价值那么下一步该怎么走。

显然tree[i].sum和pson[i]是从i的各个儿子转移得到的。

那么先做一遍DFS计算出tree[i].sum， 再排序。

然后贪心，从价值最大的那个结点开始选，从大到小。

选的时候，要把他中途经过的结点全部屏蔽（就是说被屏蔽的结点接下来不能被选了）。

当选的次数到达k或者没有结点可选时那就停止。

具体可以看我的代码。

#include <bits/stdc++.h>

using namespace std;

#define REP(i,n)                for(int i(0); i <  (n); ++i)
#define rep(i,a,b)              for(int i(a); i <= (b); ++i)
#define dec(i,a,b)              for(int i(a); i >= (b); --i)
#define for_edge(i,x)           for(int i = H[x]; i; i = X[i])

#define LL      long long

const int N     =    100000      +       10;
const int M     =    10000       +       10;
const int A     =    30          +       1;

struct Node{
    LL sum; int id;
    friend bool operator < (const Node &a, Node &b){
        return a.sum > b.sum;
    }
} tree[N];

int E[N << 2], X[N << 2], H[N << 2], pson[N];
bool isleaf[N], v[N], vc[N];
LL a[N], c[N], ans;
int T, n, k, et, x, y, num;
int Case = 0;
int cnt = 0;

bool cmp(LL a, LL b){ return a > b;}

inline void addedge(int a, int b){
    E[++et] = b, X[et] = H[a], H[a] = et;
}

void dfs(int x, int fa){
    LL cnt = 0, tot = 0, w = 0;
    tree[x].sum = a[x];
    for_edge(i, x) if (E[i] != fa){
        int u = E[i]; ++cnt;
        dfs(u, x);
        if (tree[u].sum > tot){ tot = tree[u].sum; w = u;}    
    }
    pson[x] = w;
    if (cnt == 0)  isleaf[x] = true;    
    else tree[x].sum += tree[w].sum;
}
        
void tag(int x){
    v[x] = false;
    if (!isleaf[x]) tag(pson[x]);
}

int main(){

    scanf("%d", &T);
    while (T--){
        et = 0; num = 0;
        scanf("%d%d", &n, &k);
        memset(H, 0, sizeof H);
        memset(c, 0, sizeof c);
        memset(pson, 0, sizeof pson);
        memset(vc, false, sizeof vc);
        memset(isleaf, false, sizeof isleaf);
        rep(i, 1, n) scanf("%lld", a + i);
        rep(i, 1, n - 1){
            scanf("%d%d", &x, &y);
            addedge(x, y); addedge(y, x);
            vc[y] = true;
        }
        rep(i, 1, n) tree[i].id = i;
        dfs(1, 0);
        sort(tree + 1, tree + n + 1);
        memset(v, true, sizeof v);
        cnt = 0; ans = 0;
        rep(i, 1, n) if (v[tree[i].id] && cnt < k){
            ans += tree[i].sum;
            tag(tree[i].id);
            ++cnt;
        }

        printf("Case #%d: %lld
", ++Case, ans);
        
    }

    return 0;

}