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;
}