1007. Red-black Tree (35)解题报告

思路

这道题用动态规划解决。

建立两张根分别为黑色和红色（除了根颜色外，其他都满足红黑树条件）的红黑树表，纵轴代表结点数，横轴代表黑色高度，其中元素代表树的种类数。

对于n个结点、黑色高度为h的常规红黑树，kb(n, h) = kb(i, h - 1) * kb(n - i - 1, h - 1) + kb(i, h - 1) * kr(n - i - 1, h) * 2 + kr(i, h) * kr(n - i - 1, h), i = 1, 2, ... , n - 2。

对于n个结点、黑色高度为h的红根红黑树，kr(n, h) = kb(i, h - 1) * kb(n - i - 1, h - 1), i = 1, 2, 3, ..., n - 2。  这里kb代表常规红黑树的种类数，kr代表红根红黑树的种类数。

 

在解题过程中，我参阅过下面一篇博文：

1007. Red-black Tree (35)

通过画面

 

代码

#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
#include <cstdlib>
#include <cmath>

using namespace std;

const int N = 510;
const int M = 25;
const long long MOD = 1000000007;
long long matrix[N][M];
long long red[N][M];
void dp(void);
long long complete(int n, int bx); // num of nodes and hight of black
long long comred(int n, int bx);
int main(void) {
	int n, i, j;
	long long tmp;
	matrix[1][1] = red[1][1] = red[3][2] = 1;
	matrix[2][1] = 2;
	matrix[3][1] = matrix[3][2] = 1;//row represents the number of internal nodes and col represents the black height;
	dp();
	scanf("%d", &n);
	tmp = 0;
	for (i = 0; i < M - 2; i++) {
		tmp += matrix[n][i];
		tmp %= MOD;
	}
	printf("%lld", tmp);
	return 0;
}

void dp(void) {
	int i, j, low, high;
	for (i = 4; i < 501; i++) {
		low = log2(i + 1) / 2;
		high = log2(i + 1) * 2;
		for (j = low; j <= high; j++) {
			red[i][j] = comred(i - 1, j);
			matrix[i][j] = complete(i - 1, j);
		}
	}
	return;
}

long long complete(int n, int bx) {
	int i;
	long long num = 0;
	for (i = 1; i < n; i++) {
		num += matrix[i][bx - 1] * matrix[n - i][bx - 1];
		num %= MOD;
		num += red[i][bx] * matrix[n - i][bx - 1] * 2;
		num %= MOD;
		num += red[i][bx] * red[n - i][bx];
		num %= MOD;
	}
	return num;
}

long long comred(int n, int bx) {
	int i;
	long long num = 0;
	for (i = 1; i < n; i++) {
		num += matrix[i][bx - 1] * matrix[n - i][bx - 1];
		num %= MOD;
	}
	return num;
}