【codevs1378】选课——树形动规

题目传送门

这道题可以用多叉树转二叉树的方法,左子树为儿子,右子树为同个父亲的兄弟,这样的话就可以很方便地写出状态转移方程:

f[x][v]=max(f[x][v],w[x]+dp(ch[x][0],i)+dp(ch[x][1],v-i-1));

要注意的是,f[x][v]可能在之前已经计算过(比如f[ch[i][0]][4]在f[i][5,6,7...]的时候就已经计算过)因此不必再计算,否则会TLE。

剩下的就不难了,直接DFS即可,具体实现细节看代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,ch[305][2],w[305],fa,f[305][305]; 
int read(){
    int ans=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+c-'0'; c=getchar();}
    return ans*f;
}
int dp(int x,int v)
{
    if(x==0)return 0;
    if(f[x][v])return f[x][v];//DFS剪枝
    for(int i=0;i<=v-1;i++)
        f[x][v]=max(f[x][v],w[x]+dp(ch[x][0],i)+dp(ch[x][1],v-i-1));
    f[x][v]=max(f[x][v],dp(ch[x][1],v));
    return f[x][v];
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++)
    {
        fa=read();w[i]=read();
        if(ch[fa][0])ch[i][1]=ch[fa][0];
        ch[fa][0]=i;
    }
    int ans=dp(ch[0][0],m);
    printf("%d",ans);
    return 0;
}
codevs1378

 

posted @ 2017-07-22 09:51  Child-Single  阅读(164)  评论(0编辑  收藏  举报