wikioi 1163 訪问艺术馆 树形dp
递归建树,由题知该树是一棵二叉树,且除根节点外其它点的度为0或2。
dp[i][j]表示来到第i个走廊(还未走过这条走廊)还剩下j时间,能拿到最大的画的数量。
dp[i][j]=max(dp[i][j],dp[lson[i]][k]+dp[rson][last_time-k])
#include<cstdio> #include<algorithm> using namespace std; int dp[200][700]; int id=0,x,y,s; struct node { int t,l,r; int v; }tree[200]; void init(int now) { scanf("%d%d",&x,&y); tree[now].t=x*2; tree[now].v=y; if(y!=0) tree[id].l=tree[id].r=-1; else { tree[now].l=++id;init(id); tree[now].r=++id;init(id); } } int dfs(int now,int times) { if(dp[now][times]) return dp[now][times]; //记忆化 int last=times-tree[now].t; if(last<0) return 0; if(tree[now].l==-1) return dp[now][times]=min(tree[now].v,last/5); int maxn=0; for(int i=0;i<=last;i++) { maxn=max(maxn,dfs(tree[now].l,i)+dfs(tree[now].r,last-i)); /*这里会调用多次已经计算过的值*/ //计算左儿子i时间,右儿子last-i时间的最大偷画数 } return dp[now][times]=maxn; } int main() { scanf("%d",&s); init(0); printf("%d\n",dfs(0,s)); return 0; }