poj 2342、3342(树形dp)

两道差不多的题目，放在一起了

poj 2342:

题意：在一个公司中要举办一个聚会，每一个员工有一个奉献值。为了和谐规定直接上下级不能一起出席。让你找出奉献值之和最大为多少。

思路：dp[v][1]表示当前结点选，能获得的最大奉献值，dp[v][0]表示当前节点不选能获得的最大奉献值。状态转移：

dp[v][0] = max(dp[v][0], ∑max(dp[x][1], dp[x][0]))x为直接儿子

dp[v][1] = max(dp[v][1], ∑dp[x][0] + vex[v])

最后答案是max(dp[root][0], dp[root][1])

代码如下：

/**************************************************
 * Author     : xiaohao Z
 * Blog     : http://www.cnblogs.com/shu-xiaohao/
 * Last modified : 2014-05-01 20:30
 * Filename     : poj_3342.cpp
 * Description     : 
 * ************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define MP(a, b) make_pair(a, b)
#define PB(a) push_back(a)

using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<unsigned int,unsigned int> puu;
typedef pair<int, double> pid;
typedef pair<ll, int> pli;
typedef pair<int, ll> pil;

const int INF = 0x3f3f3f3f;
const double eps = 1E-6;
const int LEN = 100000+10;
vector<int> Map[LEN];
int n, rate[LEN], In[LEN], dp[LEN][2];

void dfs(int v, int fa){
    dp[v][0] = 0;dp[v][1] = rate[v];
    int tmp = 0, tmpa = 0;
    for(int i=0; i<Map[v].size(); i++){
        int x = Map[v][i];
        if(x == fa) continue;
        dfs(x, v);
        tmp += (dp[x][0]);
        tmpa += max(dp[x][1], dp[x][0]);
    }
    dp[v][1] = max(dp[v][1], tmp+rate[v]);
    dp[v][0] = max(dp[v][0], tmpa);
}

int main()
{
//    freopen("in.txt", "r", stdin);

    int a, b;
    while(scanf("%d", &n)!=EOF && n){
        memset(In, 0, sizeof In);
        memset(dp, 0, sizeof dp);
        for(int i=0; i<n; i++) Map[i].clear();
        for(int i=0; i<n; i++){
            scanf("%d", &rate[i]);
        }
        for(int i=0; i<n-1; i++){
            scanf("%d%d", &a, &b);
            a--, b--;
            In[a] ++;
            Map[b].PB(a);
        }
        int root;
        for(int i=0; i<n; i++){
            if(In[i] == 0) {root = i;break;}
        }
        dfs(root, -1);
        printf("%d
", max(dp[root][1], dp[root][0]));
    }
    return 0;
}

 poj 3342:

代码如下：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
#define MP(a, b) make_pair(a, b)
#define PB(a) push_back(a)
using namespace std;

const int LEN = 1010;
vector<int> Map[LEN];
map<string, int> mp;
int n, top, dp[LEN][2], is[LEN][2];

int ch(string s){
    if(!mp.count(s)) mp[s] = top++;
    return mp[s];
}

void dfs(int v, int fa){
    int tmp = 0,tmpa = 0;
    dp[v][1] = 1;dp[v][0] = 0;
    for(int i=0; i<Map[v].size(); i++){
        int x = Map[v][i];
        if(x != fa){
            dfs(x, v);
            if(dp[x][0] == dp[x][1]) is[v][0] = 1;
            if(dp[x][0] > dp[x][1]){
                dp[v][0] += dp[x][0];
                is[v][0] |= is[x][0];
            }else {
                dp[v][0] += dp[x][1];
                is[v][0] |= is[x][1];
            }
            dp[v][1] += dp[x][0];
            is[v][1] |= is[x][0];
        }
    }
}

int main()
{
  //  freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    
    string a, b;
    while(cin >> n && n){
        top = 0;mp.clear();
        for(int i=0; i<LEN; i++) Map[i].clear();
        memset(dp, 0, sizeof dp);
        memset(is, 0, sizeof is);
        cin >> a;
        ch(a);
        for(int i=0; i<n-1; i++){
            cin >> a >> b;
            Map[ch(b)].PB(ch(a));
        }
        dfs(0, -1);
        int tag = 0;
        if(dp[0][1] == dp[0][0]) tag = 1;
        if(dp[0][1] > dp[0][0]){
            cout << dp[0][1] << ' ';
            if(is[0][1]) cout << "No" << endl;
            else cout << "Yes" << endl;
        }else{
            cout << dp[0][0] << ' ';
            if(is[0][0] | tag) cout << "No" << endl;
            else cout << "Yes" << endl;
        }
    }
    return 0;
}