hdu5242(贪心&记忆化搜索)

先跑一遍dfs处理出每个点到跟能获得的最大值。然后按照这个值排序，按照这个顺序再去跑第二遍dfs算出这个点 到跟的价值，用过的点直接返回0，把结果保存在数组里，再次排序，取前K个就是答案。

#include<cstdio>#include<vector>#include<cstring>#include<algorithm>using namespace std;const int maxn = 100000+10;vector<int>G[maxn];struct node{    int id;    long long len;    bool operator < (const node& T) const{        return len>T.len;    }    node(){}}p[maxn];int vis[maxn];long long w[maxn];long long dfs1(int u){    if(vis[u]) return p[u].len;    vis[u] = 1;    int len = G[u].size();    p[u].len = w[u];    for(int i=0;i<len;i++)    {        int v = G[u][i];        p[u].len += dfs1(v);    }    return p[u].len;}long long c[maxn];long long dfs2(int u){    if(vis[u]) return 0;    vis[u] = 1;    long long ret = w[u];    int len = G[u].size();    for(int i=0;i<len;i++)    {        int v = G[u][i];        ret += dfs2(v);    }    return ret;}int main(){    int cases,n,k,u,v,t=1;    scanf("%d",&cases);    while(cases--)    {        scanf("%d%d",&n,&k);        for(int i=0;i<=n;i++) G[i].clear();        for(int i=1;i<=n;i++) scanf("%I64d",&w[i]);        for(int i=1;i<n;i++)        {            scanf("%d%d",&u,&v);            G[v].push_back(u);        }        memset(vis,0,sizeof(vis));        for(int i=1;i<=n;i++) p[i].id = i;        for(int i=1;i<=n;i++)            if(!vis[i]) dfs1(i);        sort(p+1,p+1+n);        memset(vis,0,sizeof(vis));        for(int i=1;i<=n;i++)            c[i] = dfs2(p[i].id);        sort(c+1,c+1+n); long long ans=0;        for(int i=n,j=0;j<k&&i>=1;j++,i--)            ans += c[i];        printf("Case #%d: ",t++);        printf("%I64d\n",ans);    }    return 0;}