P1270 “访问”美术馆（树形dp）

题目连接：https://www.luogu.com.cn/problem/P1270

 

思路：

一开始想错了，以为是树上背包，后来发现每个非叶子结点会存在时间分配不同的问题，所以只要枚举每个非叶子结点对两棵子树分配的时间多少就好了。

 

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 2005;
struct Node
{
    int w1,w2;
}cur[N<<2];
int tim,dp[N][605];
void rd(int rt)
{
    scanf("%d",&cur[rt].w1); scanf("%d",&cur[rt].w2);
    cur[rt].w1 <<= 1;
    if(cur[rt].w2 == 0){
        rd(rt<<1);
        rd(rt<<1|1);
    }
}
int dfs(int rt,int t)
{
    t -= cur[rt].w1;
    if(t <= 0) return 0;
    if(~dp[rt][t]) return dp[rt][t];
    if(cur[rt].w2 != 0){
        return min(cur[rt].w2 , t/5);
    }
    int ans = 0;
    for(int i=0;i<=t;i++){
        ans = max(ans,dfs(rt<<1,i) + dfs(rt<<1|1,t-i));
    }
    dp[rt][t] = ans;
    return ans;
}
int main(void)
{
    memset(dp,-1,sizeof dp);
    scanf("%d",&tim); tim--;
    rd(1);
    printf("%d\n",dfs(1,tim));
    return 0;
}