题目描述
圣诞特别礼物挂在一棵圣诞树上,这棵树有n层,每层有一件礼物,每件礼物都有一个价值,有的礼物还有一些连结线,与下层的礼物相连,领取礼物的规则如下:任选一件礼物,它的下面如果有连结线,则可以继续取它连结的礼物,以此类推,直至取到没有连结线的礼物才结束,你如果是第一个去取,怎样取才能获得最大的价值呢?请你编一程序解决这一问题。
输入格式
输入文件tree.in的第一行只有一个数据n(n<=100),表示有n层礼物,以下有n行数据,分别表示第1-n层礼物的状态,每行至少由一个数据构成,且第一个数据表示该礼物的价值,后面的数据表示它与哪些层的礼物相连,如果每行只有一个数据则说明这层礼物没有与下层礼物相连,每个数的大小均不超过10000。
输出格式
输出文件tree.out也只有一个数,表示获得的取大价值。
样例输入
312 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;
}