BZOJ P3227红黑树

woc

直接树形DP嘛

然后用f[i][j][0/1]表示包含i个前端节点，高度为j根为红或黑时的方案数

然后随便转移一下就可以了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n;
int f[5010][35][2];
int maxans,minans;
int main(){
    cin>>n;
    maxans=-0x7fffffff;
	minans=0x7fffffff;
    memset(f,0x3f,sizeof f);
    f[1][1][0]=1;
	f[1][1][1]=0;
	f[2][1][1]=1;
    for(int i=1;i<=n;i++){
    	for(int j=1;j<=i;j++){
    		if(1<<j>n<<2){
    			break;
			}else{
				for(int k=0;k<=i-2;k++){
                    f[i][j][0]=min(f[i][j][0],f[k][j-1][1]+f[i-k-1][j-1][1]+1);
                    f[i][j][1]=min(f[i][j][1],min(f[k][j-1][1]+f[i-k-1][j-1][1],
                    min(f[k][j][0]+f[i-k-1][j][0],f[k][j][0]+f[i-k-1][j-1][1])));
                }
			}
		}
	}
    for(int i=0;i<=30;i++){
    	minans=min(min(minans,f[n][i][0]),f[n][i][1]);
	}
    printf("%d\n",minans);
    memset(f,-0x3f,sizeof f);
    f[1][1][0]=1;
	f[1][1][1]=0;
	f[2][1][1]=1;
    for(int i=1;i<=n;i++){
    	for(int j=1;j<=i;j++){
        	if(1<<j>n<<2){
            	break;
			}else{
				for(int k=0;k<=i-2;k++){
                    f[i][j][0]=max(f[i][j][0],f[k][j-1][1]+f[i-k-1][j-1][1]+1);
                    f[i][j][1]=max(f[i][j][1],max(f[k][j-1][1]+f[i-k-1][j-1][1],
                    max(f[k][j][0]+f[i-k-1][j][0],f[k][j][0]+f[i-k-1][j-1][1])));
                }
			}
		}
	}
    for(int i=0;i<=30;i++){
    	maxans=max(max(maxans,f[n][i][0]),f[n][i][1]);
	}
    printf("%d\n",maxans);
    return 0;
}