洛谷P2585 [ZJOI2006]三色二叉树（树形dp）

传送门

设$dp[u][i]$表示点$u$颜色为$i$时最多（最少）的绿点个数（这里用$0$表示绿点）

然后直接用树形dp就可以了

记得把情况讨论清楚

 //minamoto
 #include<bits/stdc++.h>
 #define inf 0x3f3f3f3f
 using namespace std;
 template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,:;}
 template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,:;}
 const int N=5e5+;
 char s[N];int n,cnt,L[N],R[N],dp[N][];
 void dfs(int u){
     ++cnt;
     switch(s[u]){
         case :break;
         case :L[u]=cnt,dfs(cnt);break;
         case :L[u]=cnt,dfs(cnt),R[u]=cnt,dfs(cnt);break;
     }
 }
 #define ls L[u]
 #define rs R[u]
 void dfsmax(int u){
     if(!ls) return (void)(dp[u][]=,dp[u][]=dp[u][]=);
     dfsmax(ls);
     if(rs) dfsmax(rs);
     dp[u][]=dp[u][]=dp[u][]=;
     if(rs){
         cmax(dp[u][],dp[ls][]+dp[rs][]+),
         cmax(dp[u][],dp[ls][]+dp[rs][]+);

         cmax(dp[u][],dp[ls][]+dp[rs][]),
         cmax(dp[u][],dp[ls][]+dp[rs][]);

         cmax(dp[u][],dp[ls][]+dp[rs][]),
         cmax(dp[u][],dp[ls][]+dp[rs][]);
     }else{
         cmax(dp[u][],dp[ls][]+),cmax(dp[u][],dp[ls][]+);
         cmax(dp[u][],dp[ls][]),cmax(dp[u][],dp[ls][]);
         cmax(dp[u][],dp[ls][]),cmax(dp[u][],dp[ls][]);
     }
 }
 void dfsmin(int u){
     if(!ls) return (void)(dp[u][]=,dp[u][]=dp[u][]=);
     dfsmin(ls);
     if(rs) dfsmin(rs);
     dp[u][]=dp[u][]=dp[u][]=inf;
     if(rs){
         cmin(dp[u][],dp[ls][]+dp[rs][]+),
         cmin(dp[u][],dp[ls][]+dp[rs][]+);

         cmin(dp[u][],dp[ls][]+dp[rs][]),
         cmin(dp[u][],dp[ls][]+dp[rs][]);

         cmin(dp[u][],dp[ls][]+dp[rs][]),
         cmin(dp[u][],dp[ls][]+dp[rs][]);
     }else{
         cmin(dp[u][],dp[ls][]+),cmin(dp[u][],dp[ls][]+);
         cmin(dp[u][],dp[ls][]),cmin(dp[u][],dp[ls][]);
         cmin(dp[u][],dp[ls][]),cmin(dp[u][],dp[ls][]);
     }
 }
 int main(){
 //    freopen("testdata.in","r",stdin);
     scanf("%s",s+),n=strlen(s+);
     for(int i=;i<=n;++i) s[i]-='';
     cnt=,dfs();
     dfsmax(),printf("%d ",max(dp[][],max(dp[][],dp[][])));
     dfsmin(),printf("%d\n",min(dp[][],min(dp[][],dp[][])));
     return ;
 }