BZOJ 1864 三色二叉树 - 树型dp

传送门

题目大意：

给一颗二叉树染色红绿蓝，父亲和儿子颜色必须不同，两个儿子颜色必须不同，问最多和最少能染多少个绿色的、

题目分析：

裸的树型dp：\(dp[u][col][type]\)表示u节点染为col(0-绿色，1-红色，2-蓝色)，当前求的是type（0-最小，1-最大）解。
然后最后输出最大为\(max(dp[1][0][1], dp[1][1][1], dp[1][2][1])\)，最小为\(min(dp[1][0][0], dp[1][1][0], dp[1][2][0])\)。

code

#include<bits/stdc++.h>
using namespace std;

const int N = 500005, oo = 0x3f3f3f3f;
vector<int> adj[N];
int tot, dp[N][3][2];

inline void read(int k){
    int t, v; char tt; scanf("%c", &tt);
    t = tt - '0';
    for(int i = 1; i <= t; i++){
        adj[k].push_back(v = ++tot);
        read(v);
    }
}

inline int DP(int u, int f, int col, int type){
    if(dp[u][col][type] != -1) return dp[u][col][type];
    dp[u][col][type] = col == 0 ? 1 : 0;
    if(!adj[u].size()) return dp[u][col][type];
    
    int col1, col2;   //另外两种 
    if(col == 0) col1 = 1, col2 = 2;
    else if(col == 1) col1 = 0, col2 = 2;
    else col1 = 0, col2 = 1;
    
    if(adj[u].size() == 1){   //只有一个儿子 
        int v = adj[u][0];
        if(type == 1)
            dp[u][col][type] += max(DP(v, u, col1, type), DP(v, u, col2, type));
        else dp[u][col][type] += min(DP(v, u, col1, type), DP(v, u, col2, type));
    }
    else{                   //只有两个儿子 
        int v1 = adj[u][0], v2 = adj[u][1];
        int t1 = DP(v1, u, col1, type) + DP(v2, u, col2, type);
        int t2 = DP(v1, u, col2, type) + DP(v2, u, col1, type);
        if(type == 1)
            dp[u][col][type] += max(t1, t2);
        else dp[u][col][type] += min(t1, t2);
    }
    return dp[u][col][type];
}

int main(){
    read(tot = 1);
    memset(dp, -1, sizeof dp);
    int ans1 = -1, ans2 = oo;
    for(int i = 0; i < 3; i++)
        ans1 = max(ans1, DP(1, 0, i, 1)), ans2 = min(ans2, DP(1, 0, i, 0));
    printf("%d %d", ans1, ans2);
    return 0;
}

转载于:https://www.cnblogs.com/CzYoL/p/7704377.html