hdu 4616 Game(树形DP)

题意：
一颗有n个节点的树，每个节点有val值和c(1-有陷阱，0-无陷阱)值，给出最大可踩陷阱数k。在树中任取一点作为起点，经过某点就取得该点的val值，
踩到第k个陷阱后马上停止，而且不能走已走过的点。
求最大累计val值。
暴力方法：每点进行一次深搜求最大值。（这样竟然可以过= =，703ms）
树形DP：
子问题：最大值转化为求一个节点某子树的最大值和树的其他部分的最大值和。
状态：dp[u][j]表示从以u为根，从u子树中某点走到u经过j个陷阱时的最大val值。
难点：与“不超过 k 个求最大的长度”不同的是，该题是达到k个就马上停止
所以增加一维表示方向，即dp[u][j][0]表示从非陷阱点走到u，dp[u][j][1]表示从陷阱点走到u
更详细解释，见代码注释

暴力：

1 #include <cstdio>
2 #include <cstring>
3 #define N 50005
4 #define INF 99999999
5
6 struct Edge
7 {
8 int e, next;
9 }edge[N*2];

int head
, eNum;
int c
, val
, n, k, ans, dp
[4][2];

void init()
{
memset(head, -1, sizeof(head));
eNum = 0; ans = 0;
for(int i=0; i<n; i++)
{
for(int j=0; j<4; j++)
{
for(int l=0; l<2; l++)
dp[i][j][l] = -INF;
}
}
}

void AddEdge(int u, int v)
{
edge[eNum].e = v;
edge[eNum].next = head[u];
head[u] = eNum++;
}

int max2(int x, int y)
{
return x > y ? x : y;
}

void dfs(int u, int fa)
{
dp[u][c[u]][c[u]] = val[u]; //初始化dp[u][0][0] 或 dp[u][1][1]
for(int i=head[u]; i!=-1; i=edge[i].next)
{
int v = edge[i].e;
if(v==fa) continue;
//先求从子节点子树到子节点的最大值
dfs(v, u);
//用u节点子树到u节点的最大值 和 从子节点子树到子节点的最大值 和 更新ans
for(int j=0; j<=k; j++) //-u
{
for(int l=0; l+j<=k; l++)//-v
{
ans = max2(ans, dp[u][j][1]+dp[v][l][1]); // 1 -> 1
if(j+l!=k) ans = max2(ans, dp[u][j][0]+dp[v][l][0]); // 0 -> 0
ans = max2(ans, dp[u][j][0]+dp[v][l][1]); // 0 -> 1 dp[v][0][1] 为-INF
ans = max2(ans, dp[u][j][1]+dp[v][l][0]); // 0 -> 1 dp[u][0][1] 为-INF
}
}
for(int j=0; j<k; j++)
{
dp[u][j+c[u]][1] = max2(dp[u][j+c[u]][1],dp[v][j][1]+val[u]); //用子节点更新u点
dp[u][j+c[u]][0] = max2(dp[u][j+c[u]][0],dp[v][j][0]+val[u]);
}
if(c[u]==0) dp[u][k][1] = max2(dp[u][k][1], dp[v][k][1]+val[u]); //这里漏了，存在ans = dp[v1][0][0] + dp[u][k][1]的可能
}
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&k);
for(int i=0; i<n; i++)
scanf("%d%d",&val[i],&c[i]);
init();
int u, v;
for(int i=0; i<n-1; i++)
{
scanf("%d%d",&u,&v);
AddEdge(u, v);
AddEdge(v, u);
}
dfs(0,-1);
printf("%d\n",ans);
}
return 0;
}View Code