[HDU] 1520 Anniversary party 入门树形DP

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1520

方法： 领导选择出现，则其直接下属不能出现，领导选择不出现，则其直接下属可出现可不出现，设F(P,0)为一个员工不出现能得的最大值，F(P,1)为一个员工要出现能得的最大值

　　　　　　{

　　　　　　sum({F(S,0)|S是p的直接下属})+p的有趣度;P为非叶子

F（P，1）=

　　　　　　p的有趣度;P为叶子

　　　　　　}，

　　　　　　{

　　　　　　sum({ min(F(S,1),F(S,0)) |S是p的直接下属 });P为非叶子

F（P，0）=

　　　　　　0;P为叶子

　　　　　　}

原来问题的解是min(F(root,1),F(root,0)).

注意输入不一定是树结构，有可能是森林，所以要把所有的树都DP一次，把结果累加起来。

代码：

#include <iostream>
#include <queue>
#include<algorithm>
using namespace std;
int n,m;
struct Arc
{
int vetex;
Arc* nextArc;
};
struct Node
{
int x;
int inDegree;
Arc* firstArc;
int value;
int in;
int notIn;
};
Node nodes[6001];
int myMax(int x,int y)
{
return x>y ? x:y;
}
void DFSDP(int root)
{
Arc* arc = nodes[root].firstArc;
int sonCount=0,v;
while(arc!=NULL)
{
v=arc->vetex;
DFSDP(v);
nodes[root].in+=nodes[v].notIn;
nodes[root].notIn+=myMax(nodes[v].notIn,nodes[v].in);
arc=arc->nextArc;
sonCount++;
}
if(sonCount==0)
{
nodes[root].in=nodes[root].value;
nodes[root].notIn=0;
}
}
void createArc(int s,int e)
{
Arc* arc = (Arc*)malloc(sizeof(Arc));
arc->vetex=e;
if(nodes[s].firstArc==NULL)
arc->nextArc=NULL;
else
arc->nextArc=nodes[s].firstArc;
nodes[s].firstArc = arc;
}
void main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d",&nodes[i].value);
nodes[i].firstArc=NULL;
nodes[i].notIn=nodes[i].inDegree=0;
nodes[i].in=nodes[i].value;
}
int emp,boss;
while(scanf("%d %d",&emp,&boss) && !(emp==0 && boss==0))
{
nodes[boss].x = boss;
nodes[emp].x = emp;
createArc(boss,emp);
nodes[emp].inDegree++;
}
int re=0;
for(int i =1;i<=n;i++)
{
if(nodes[i].inDegree==0)
{
DFSDP(i);
re+=myMax(nodes[i].in,nodes[i].notIn);
}
}
cout<<re<<endl;
}
}

感想：模板题