首页 > 代码库 > BZOJ1864[ZJOI2006]三色二叉树[树形DP]

BZOJ1864[ZJOI2006]三色二叉树[树形DP]

2024-08-14 00:21:10   220人阅读

1864: [Zjoi2006]三色二叉树

Time Limit: 1 Sec  Memory Limit: 64 MB
Submit: 773  Solved: 548
[Submit][Status][Discuss]

Description

技术分享

Input

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

Output

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

Sample Input

1122002010

Sample Output

5 2

HINT

 

Source

Day1

先建树

d[i][0/1]表示以i为根的子树i绿色1其他0下最多几个点绿色 

三个颜色转移就是父子三人两个其他一个绿色

见代码

/**************************************************************    Problem: 1864    User: thwfhk    Language: C++    Result: Accepted    Time:68 ms    Memory:13884 kb****************************************************************/#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int N=5e5+5,INF=1e9;int n,mx=-INF,mn=INF;char s[N];struct node{    int ls,rs;    node():ls(0),rs(0){}}tree[N];int cnt=0,pos=1;int build(){    int u=++cnt,num=s[pos]-‘0‘;    pos++;    if(num>=1) tree[u].ls=build();    if(num==2) tree[u].rs=build();    return u;}int d[N][2],f[N][2];void dp(int u){//printf("dp %d\n",u);    if(u==0) return;    int ls=tree[u].ls,rs=tree[u].rs;    dp(ls);dp(rs);    d[u][0]=max(d[ls][0]+d[rs][1],d[ls][1]+d[rs][0]);    f[u][0]=min(f[ls][0]+f[rs][1],f[ls][1]+f[rs][0]);    d[u][1]=d[ls][0]+d[rs][0]+1;    f[u][1]=f[ls][0]+f[rs][0]+1;}int main(int argc, const char * argv[]) {    scanf("%s",s+1);    n=strlen(s+1);    build();    dp(1);    mx=max(d[1][0],d[1][1]);    mn=min(f[1][0],f[1][1]);    printf("%d %d",mx,mn);    return 0;}

 

BZOJ1864[ZJOI2006]三色二叉树[树形DP]


联系
我们