其他OJ 树型DP 技能树(未通过)

http://www.cqoi.net:2012/JudgeOnline/problem.php?id=1380

题意什么的都在里面，中文题目不用解释

个人感觉是数据的问题，因为网上找遍了代码都不能通过，算了还是不纠结了

代码写得也不好，无心机改了

/*
dp思想:dp[rt][p]，当前节点rt，有p点，能获得的最大价值
1.可以用一部分来升级当前的这个技能，升的级数不确定，但升级需要花费点数，花费的点数
  c<=p
2.不升级当前,全部用于升级子树和兄弟。但注意，若不升级当前节点而且当前节点等级是0，那么
  其子树不能升级，只能升级其兄弟
*/

#include <cstdio>
#include <cstring>
#define    N 25
#define M 25
#define LEN 25
#define MAX 1010

char skill[N][LEN];
char parent[N][LEN];
int fc[N],rb[N];
int level[N];
int cost[N][M];
int val[N][M];
int start[N];
int sn;
int P;
int dp[N][MAX];

int search(char *name)
{
    if(name[0]=='\0') return 0;
    for(int i=1; i<=sn; i++)
        if(!strcmp(skill[i] , name))
            return i;
    return -1;
}

void input()
{
    memset(fc,-1,sizeof(fc));
    memset(rb,-1,sizeof(rb));
    memset(level,0,sizeof(level));
    memset(cost,0,sizeof(cost));
    memset(val,0,sizeof(val));
    memset(skill,0,sizeof(skill));

    char name[LEN];
    for(int nn=1; nn<=sn; nn++)
    {
        int i;
        for(i=0; (name[i]=getchar())!='\n'; ) i++;
        name[i]='\0'; 
        strcpy(skill[nn] , name);
        //printf("%s\n",name);

        for(i=0; (name[i]=getchar())!='\n'; ) i++;
        name[i]='\0'; 
        strcpy(parent[nn] , name);
        //printf("%s\n",name);

        scanf("%d",&level[nn]);
        for(i=0; i<level[nn]; i++) 
            scanf("%d",&cost[nn][i]);
        for(i=0; i<level[nn]; i++)
            scanf("%d",&val[nn][i]);
        getchar();
    }

    scanf("%d",&P);
    for(int i=1; i<=sn; i++) 
        scanf("%d",&start[i]);

    for(int nn=1; nn<=sn; nn++)
    {
        int par=search(parent[nn]);
        int firson=fc[par];
        fc[par]=nn; rb[nn]=firson;
    }
}

int max(int x ,int y)
{
    return x>y?x:y;
}

int cal(int rt ,int s ,int e)
{
    int ans=0;
    for(int i=s; i<=e; i++)
        ans += cost[rt][i];
    return ans;
}

int sum(int rt ,int s,int e)
{
    int ans=0;
    for(int i=s; i<=e; i++)
        ans += val[rt][i];
    return ans;
}

int dfs(int rt , int p)
{
    if(rt<0) return 0;
    if(p==0) return dp[rt][p]=0;
    if(dp[rt][p]!=-1) return dp[rt][p];
    dp[rt][p]=0;

    for(int w=0; w<=p; w++) 
    {//先不给当前节点升级，将点数全部用于升级子树或兄弟
        int s1=0 ,s2=0;
        int firson=fc[rt] , bro=rb[rt];
        if(start[rt]) 
            s1=dfs(firson,p-w);
        s2=dfs(bro,w);
        dp[rt][p] = max(dp[rt][p] , s1+s2);
    }
    int c,v;
    for(int i=start[rt]; i<level[rt]; i++) //选择给当前节点升级,从它最初的等级开始
    {
        c=cal(rt,start[i],i); //升这么多级需要的点数
        v=sum(rt,start[i],i); //升这么多级能获得的价值
        if(c>p) break; //超过当前点数了直接跳出
        for(int w=0; w<=p-c; w++) //在剩下的p-c点数中分给孩子和兄弟
        {
            int s1=0 ,s2=0;
            int firson=fc[rt] , bro=rb[rt];
            s1=dfs(firson,p-c-w);
            s2=dfs(bro,w);

            dp[rt][p] = max(dp[rt][p] , s1+s2+v);
        }
    }
    return dp[rt][p];
}

void solve()
{
    memset(dp,-1,sizeof(dp));
    dfs(fc[0],P);
    printf("%d\n",dp[fc[0]][P]);
}

int main()
{
    while(scanf("%d",&sn)!=EOF)
    {
        getchar();
        input();
        solve();
    }
    return 0;
}