HDU 4616 Game

来源：互联网发布：覆盖法贪心算法编辑：程序博客网时间：2024/05/29 06:43

树状DP，这题我写了2个Dfs来做，第一个以任意一点（我用的0节点）作为根节点，求出以每个点作为根节点的子树上机会为1~C时各自的最大值，然后第二次Dfs用父亲节点来更新儿子节点，求出每个点作为根节点能获得的最大值，做了之后看了看别人的题解，貌似只用一个Dfs就够了，而且代码量也比较少，等会儿再去看看吧。

#include <cstdio>#include <algorithm>using namespace std;const int N = 50005;struct node{       int val , fg;       int maxval[5];//保存机会为1~C时的最大值        int maxpos[5];//保存取得最大值的路径        int smaxval[5];//保存机会为1~C时的次最大值        int smaxpos[5];//保存取得次最大值的路径 }tp[N];struct Node{       int ev , next;}tree[N*2];int n , c , tot , father[N];void Addedge(int s , int e){     tree[tot].ev = e;     tree[tot].next = father[s];     father[s] = tot++;}void Dfs(int s , int fa){     int i , j , vs = 0;     memset(tp[s].maxval,0,sizeof tp[s].maxval);     memset(tp[s].smaxval,0,sizeof tp[s].smaxval);     for (i = father[s] ; i != -1 ; i = tree[i].next)     {         int v = tree[i].ev;         if (v == fa)continue;         vs = 1;         Dfs(v,s);         int cnt = tp[s].fg;         for (j = 1 ; j <= c ; j++)         {             if (tp[s].maxval[j] < tp[v].maxval[j-cnt]+tp[s].val)             {                                 tp[s].smaxval[j] = tp[s].maxval[j];                                 tp[s].smaxpos[j] = tp[s].maxpos[j];                                 tp[s].maxval[j] = tp[v].maxval[j-cnt]+tp[s].val;                                 tp[s].maxpos[j] = v;             }             else if (tp[s].smaxval[j] < tp[v].maxval[j-cnt]+tp[s].val)             {                  tp[s].smaxval[j] = tp[v].maxval[j-cnt]+tp[s].val;                  tp[s].smaxpos[j] = v;             }         }     }     if (vs)return;     for (j = 1 ; j <= c ; j++)     tp[s].maxval[j] = tp[s].val;}void Dfs1(int s , int fa){     int i , j;     for (i = father[s] ; i != -1 ; i = tree[i].next)     {         int v = tree[i].ev;         if (v == fa)continue;         int cnt = tp[s].fg;         for (j = 1 ; j <= c ; j++)         {             if (tp[v].maxval[j] < tp[s].maxval[j-cnt]+tp[v].val && tp[s].maxpos[j-cnt] != v)             {                                 tp[v].smaxval[j] = tp[v].maxval[j];                                 tp[v].smaxpos[j] = tp[v].smaxpos[j];                                 tp[v].maxval[j] = tp[s].maxval[j-cnt]+tp[v].val;                                 tp[v].maxpos[j] = s;             }             else if (tp[v].maxval[j] < tp[s].smaxval[j-cnt]+tp[v].val && tp[s].smaxpos[j-cnt] != v)             {                  tp[v].smaxval[j] = tp[v].maxval[j];                  tp[v].smaxpos[j] = tp[v].smaxpos[j];                  tp[v].maxval[j] = tp[s].smaxval[j-cnt]+tp[v].val;                      tp[v].maxpos[j] = s;             }             else if (tp[v].smaxval[j] < tp[s].maxval[j-cnt]+tp[v].val && tp[s].maxpos[j-cnt] != v)             {                  tp[v].smaxval[j] = tp[s].maxval[j-cnt]+tp[v].val;                  tp[v].smaxpos[j] = s;             }             else if (tp[v].smaxval[j] < tp[s].smaxval[j-cnt]+tp[v].val && tp[s].smaxpos[j-cnt] != v)             {                   tp[v].smaxval[j] = tp[s].smaxval[j-cnt]+tp[v].val;                  tp[v].smaxpos[j] = s;             }         }         Dfs1(v,s);     }}int main(){    int t , i;    scanf("%d",&t);    while (t--)    {          memset(father , -1 , sizeof father);          tot = 0;          scanf("%d%d",&n,&c);          for (i = 0 ; i < n ; i++)scanf("%d%d",&tp[i].val,&tp[i].fg);          for (i = 1 ; i < n ; i++)          {              int s , e;              scanf("%d%d",&s,&e);              Addedge(s,e);              Addedge(e,s);          }          Dfs(0,-1);          Dfs1(0,-1);          int MAX = 0;          for (i = 0 ; i < n ; i++)MAX = max(MAX , tp[i].maxval[c]);          printf("%d\n",MAX);    }        return 0;}