没有上司的舞会
题目描述
某大学有N个职员,编号为1~N。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入格式
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0
输出格式
输出最大的快乐指数。
输入输出样例
输入
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
输出
5
分析
这题我们可以用动态规划(DP)来做
我们要思考动态转移方程
因为有两种情况(取和不取),所以我们可以得出:
但是,这是树状动规,肯定要有父节点和子节点的图,那么,我们可以用连接表来做(预处理)
struct stu//可以用结构体
{
int x,to,next;
}b[6005];
void add(int x,int y)//链表建立
{
b[++s].x=x;//他的上司
b[s].to=y;//自己
b[s].next=head[x];//下一个
head[x]=s;//为了方便下一个找自己
}
最后就会这样
做好这些以后,再找到没有上司的"老大",由这个点递归到最深处,最后求出结果
但是,结果也有两种,一种是不取自己,另一种是取自己
所以,还要再求一个最大值
cout<<max(c[k][0],c[k][1]);//不取或取中找最大
注意:我这里1是取这个点,0是不取这个点
AC代码
#include<iostream>
using namespace std;
int s,m,x,y,k,f[6005],c[6005][2],head[6005],h[6005];
struct stu//可以用结构体
{
int x,to,next;
}b[6005];
void add(int x,int y)//链表建立
{
b[++s].x=x;//他的上司
b[s].to=y;//自己
b[s].next=head[x];//下一个
head[x]=s;//为了方便下一个找自己
}
void dp(int n)
{
c[n][1]=h[n];//初始值
for(int i=head[n];i;i=b[i].next)//翻译为"i=head[n];while(i!=0){执行语句;i=b[i].next;}"就是从头找下去
{
dp(b[i].to);//递归到根部
c[n][1]=c[n][1]+c[b[i].to][0];//取这个点,他的儿子都不能要
c[n][0]=max(c[b[i].to][1],c[b[i].to][0])+c[n][0];//不取这个点,取价值最大的情况
}
}
int main()
{
cin>>m;
for(int i=1;i<=m;i++)
cin>>h[i];
cin>>x>>y;
while(x!=0||y!=0)
{
add(y,x);
f[x]=1;//有上司
cin>>x>>y;
}
for(int i=1;i<=m;i++)//找到"老大"
if(f[i]==0){
k=i;break;}
dp(k);
cout<<max(c[k][0],c[k][1]);//取或不取中找最大
}