BZOJ-1864: [Zjoi2006]三色二叉树 (julao都说简单的树形DP)

1864: [Zjoi2006]三色二叉树

Time Limit: 1 Sec  Memory Limit: 64 MB
Submit: 1118  Solved: 818
[Submit][Status][Discuss]

Description

Input

仅有一行，不超过500000个字符，表示一个二叉树序列。

Output

输出文件也只有一行，包含两个数，依次表示最多和最少有多少个点能够被染成绿色。

Sample Input

1122002010

Sample Output

5 2

HINT

 

Source

Day1

哇呀呀呀呀呀这么简单的树形dp竟然没想到  啊啊啊啊啊laj好菜哇

f[x][1]表示当前点涂绿他和他的子树中涂绿的个数 f[x][0]表示当前点非绿他和他的子树中涂绿的个数

然后……乱搞搞就行咯 _(:зゝ∠)_

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX=3e5+5;
int n,ans1,ans2,l[MAX],r[MAX],f[MAX][2];
void scan(int x){
    char c=getchar();
    if (c=='0') return;
    ++n;l[x]=n; scan(n);
    if (c=='2') ++n,r[x]=n,scan(n);
}
void dfs1(int x){
    if (!x) return;
    dfs1(l[x]),dfs1(r[x]);
    f[x][1]=f[l[x]][0]+f[r[x]][0]+1;
    f[x][0]=max(f[l[x]][0]+f[r[x]][1],f[l[x]][1]+f[r[x]][0]);
}
void dfs2(int x){
    if (!x) return;
    dfs2(l[x]),dfs2(r[x]);
    f[x][1]=f[l[x]][0]+f[r[x]][0]+1;
    f[x][0]=min(f[l[x]][0]+f[r[x]][1],f[l[x]][1]+f[r[x]][0]);
}
int main(){
    freopen ("tree.in","r",stdin);freopen ("tree.out","w",stdout);
    int i,j;n=1;
    scan(1);
    dfs1(1);ans1=max(f[1][0],f[1][1]);
    memset(f,0,sizeof(f));
    dfs2(1);ans2=min(f[1][0],f[1][1]);
    printf("%d %d",ans1,ans2);
    return 0;
}