没有上司的晚会没有上司的晚会 没 有 上 司 的 晚 会
Description
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
Input
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。
Output
输出最大的快乐指数。
Sample Input
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
Sample Output
5
题目大意:
有一场party,有n个人会来,每个人都有自己的happy度,部分人有自己上司(见到自己的上司就会无法发挥happy值),求最大的happy指数
解题方法:
用f[i][1](f[i][1]+=f[i的下属][0])表示第i个人去,f[i][0](f[i][0]+=max(f[i的下属][0],f[i的下属][1]))表示第i个人不去,用链表的形式存每一条关系线的上司(x),属下(y),头相等的另一条线(next),再用一个head来存每个点连接下属的线,再用递归来求出结果。
#include <cstdio>
#include <iostream>
using namespace std;
int n, k, l, pd[ 6002 ] , a[ 6002 ] , f[ 6002 ] [ 3 ] , head[ 6002 ] ;
struct abc
{ int x, y, next;
} s[ 6002 ] ;
void dp ( int dep)
{ f[ dep] [ 1 ] = a[ dep] ; for ( int i= head[ dep] ; i; i= s[ i] . next) { dp ( s[ i] . y) ; f[ dep] [ 0 ] + = max ( f[ s[ i] . y] [ 1 ] , f[ s[ i] . y] [ 0 ] ) ; f[ dep] [ 1 ] + = f[ s[ i] . y] [ 0 ] ; }
}
int main ( )
{ scanf ( "%d" , & n) ; for ( int i= 1 ; i<= n; i++ ) scanf ( "%d" , & a[ i] ) ; for ( int i= 1 ; i<= n; i++ ) { scanf ( "%d%d" , & l, & k) ; s[ i] . x= k; s[ i] . y= l; s[ i] . next= head[ k] ; head[ k] = i; pd[ l] = 1 ; } for ( int i= 1 ; i<= n; i++ ) if ( ! pd[ i] ) { dp ( i) ; printf ( "%d" , max ( f[ i] [ 1 ] , f[ i] [ 0 ] ) ) ; break ; }
}