码迷,mamicode.com
首页 > 其他好文 > 详细

【codevs1378】选课——树形动规

时间:2017-07-22 10:54:21      阅读:164      评论:0      收藏:0      [点我收藏+]

标签:opened   实现   div   main   传送门   amp   getchar   cst   方便   

题目传送门

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

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

 

【codevs1378】选课——树形动规

标签:opened   实现   div   main   传送门   amp   getchar   cst   方便   

原文地址:http://www.cnblogs.com/JKAI/p/7220417.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!