树形DP专题已经过去了一星期了。写几道题解来加强理解吧。
Anniversary party
(树形DP入门)题意:
有n个人要去参加party,但是一个人来参加了,那么他的下属就不能来。输入的时候,n行给出n个人来的价值,然后输入x,y,表示x是y的下属,x和y同时为0输入停止,表示x是y的下属。求这次party可以获得的最大价值。
思路:树形DP,dp【i】【0】表示第i个人不来,dp【i】【1】表示第i个人来。那么状态转移方程就是
dp[f][1]+=dp[z][0];
dp[f][0]+=max(dp[z][0],dp[z][1]);
z为f的下属。
AC代码:
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int mx=100010;
struct node{
int w,v,nex;
}a[mx<<1];
int dp[mx][3],h[mx],n,t;
int f[mx];
void add(int w,int v,int f)
{
a[t].w=w;
a[t].v=v;
a[t].nex=h[f];
h[f]=t++;
}
void dfs1(int f);
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(h,-1,sizeof(h));
memset(f,-1,sizeof(f));
memset(dp,0,sizeof(dp));t=0;
for(int i=1;i<=n;i++)
{scanf("%d",&dp[i][1]);}
int x,y;
while (scanf("%d%d",&x,&y)&&!(x==0&&y==0))
{ add(x,x,y);f[x]=y;}
int i=1;
while(f[i]!=-1) i=f[i];
dfs1(i);
printf("%d\n",max(dp[i][1],dp[i][0]));
}
return 0;
}
void dfs1(int f)
{
for(int i=h[f];i!=-1;i=a[i].nex)
{
int z=a[i].v;
dfs1(z);
{
dp[f][1]+=dp[z][0];
dp[f][0]+=max(dp[z][0],dp[z][1]);
}
}
}