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;  
}