POj 2342 树形DP

题目：http://poj.org/problem?id=2342
题意：经典舞会问题。
很久没做树形DP了。做起来还有点忘记了的- -。#include<bits/stdc++.h>
#define INF 1e18
#define inf 1e9
#define min(a,b) a<b?a:b
#define max(a,b) a>b?a:b
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define IOS ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std ;
typedef long long ll;
typedef unsigned long long ull;
const int _max = 6005;

int dp[_max][2];
int c[_max];
vector<int> mp[_max];
void dfs_dp(int f){
for(int i = 0 ; i < mp[f].size() ; i++){
int s = mp[f][i];
dfs_dp(s);
dp[f][1] += dp[s][0];
dp[f][0] += max(dp[s][1],dp[s][0]);
}
}
int main(){
IOS;
int n;
while(cin>>n){
memset(dp,0,sizeof(dp));
memset(c,0,sizeof(c));
for(int i = 1 ; i <= n ; i++){
cin>>dp[i][1];
mp[i].clear();
}
int u,v;
while(cin>>u>>v){
if(!u && !v) break;
c[u]++;
mp[v].push_back(u);
}
int root;
for(int i = 1 ; i <= n ; i++)
if(c[i] == 0){
root = i;
break;
}
dfs_dp(root);
int ans_max = max(dp[root][1],dp[root][0]);
cout<<ans_max<<endl;
}

return 0;
}我是用容器建树。网上看到一个人用并集合查根的方法，用一个一维数组建树做法。#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>

using namespace std;

#define maxn 6005

int n;
int dp[maxn][2],father[maxn];//dp[i][0]0表示不去，dp[i][1]1表示去了
bool visited[maxn];

void tree_dp(int node)
{
int i;
visited[node] = 1;
for(i=1; i<=n; i++)
{
if(!visited[i]&&father[i] == node)//i为下属
{
tree_dp(i);//递归调用孩子结点，从叶子结点开始dp
//关键
dp[node][1] += dp[i][0];//上司来,下属不来
dp[node][0] +=max(dp[i][1],dp[i][0]);//上司不来，下属来、不来
}
}
}

int main()
{
int i;
int f,c,root;
while(scanf("%d",&n)!=EOF)
{
memset(dp,0,sizeof(dp));
memset(father,0,sizeof(father));
memset(visited,0,sizeof(visited));
for(i=1; i<=n; i++)
{
scanf("%d",&dp[i][1]);
}
root = 0;//记录父结点
bool beg = 1;
while (scanf("%d %d",&c,&f),c||f)
{
father[c] = f;
if( root == c || beg )
{
root = f;
}
}
while(father[root])//查找父结点
root=father[root];
tree_dp(root);
int imax=max(dp[root][0],dp[root][1]);
printf("%d\n",imax);
}
return 0;
}