RQNOJ 172 圣诞树

题目描述

圣诞特别礼物挂在一棵圣诞树上，这棵树有n层，每层有一件礼物，每件礼物都有一个价值，有的礼物还有一些连结线，与下层的礼物相连，领取礼物的规则如下：任选一件礼物，它的下面如果有连结线，则可以继续取它连结的礼物，以此类推，直至取到没有连结线的礼物才结束，你如果是第一个去取，怎样取才能获得最大的价值呢？请你编一程序解决这一问题。

输入格式

输入文件tree.in的第一行只有一个数据n(n<=100),表示有n层礼物，以下有n行数据，分别表示第1-n层礼物的状态，每行至少由一个数据构成，且第一个数据表示该礼物的价值，后面的数据表示它与哪些层的礼物相连，如果每行只有一个数据则说明这层礼物没有与下层礼物相连，每个数的大小均不超过10000。

输出格式

输出文件tree.out也只有一个数，表示获得的取大价值。

样例输入

3
12 2 3
20
30

样例输出

42

读入比较恶心 其他还好 类似于数字三角形的dp

一开始也尝试过dfs自顶向下搜索 但是会TLE 太懒 没剪枝

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
string str;
int m,n,a,b,c,ans;

struct self{int x,y;}s[11111];
int first[11111],next[11111];
int w[111];
int f[111];
void makeside(int x,int y)
{
    n++;
    s[n].x=x;s[n].y=y;
    next[n]=first[x];first[x]=n;
}    

void work(string str,int i)
{
    int a,b,read=0;
    for(a=0;a<str.length()&&str[a]!=' ';a++)
    read=read*10+str[a]-'0';
    w[i]=read;
    
    for(a;a<str.length();a++)
    if(str[a]==' ')
    {
        read=0;
        for(b=a+1;b<str.length()&&str[b]!=' ';b++)
        read=read*10+str[b]-'0';
        makeside(i,read);
    }
}
        
int dfs(int i)
{
    
    if(f[i]==-1)
    for(int e=first[i];e!=-1;e=next[e])
    f[i]=max(dfs(s[e].y)+w[i],f[i]);
    f[i]=max(f[i],w[i]);
    return f[i];
}    
       
int main()
{
    memset(first,-1,sizeof(first));memset(next,-1,sizeof(next));
    memset(f,-1,sizeof(f));
    scanf("%d\n",&m);
    for(a=1;a<=m;a++)
    {
        getline(cin,str);
        work(str,a);
    }
    for(a=1;a<=m;a++)ans=max(ans,dfs(a));
    cout<<ans<<'\n';
    return 0;
}