摩天楼 DP优化

有N(≤1300)栋摩天楼,从左往右排成一行。第i栋的高度是i个单位,并且每栋摩天楼都有颜色(有些摩天楼的颜色相同),将这些摩天楼排列,有多少种排列的方法,使得从左往右看去,看到不多不少刚好L栋摩天楼,答案mod 1000000009。

这里定义一下“看到”,首先若某栋摩天楼左边有比它高的,那就看不到。如果从左往右依次连续看到的两栋(或以上)摩天楼颜色相同,会误认为是一栋。

由于这个关系到次序问题,所以DP的顺序可以按照从大到小的顺序来,这里只需要把读入的顺序倒过来(读入是从小到大,下面的描述第i栋的高度是n - i + 1)。

记f(j,i)为能够看到j栋摩天楼,并且放置了前i栋摩天楼后,第一栋看到的楼就是第i栋楼。求出f(j,i)需要枚举前一栋看到的楼是什么,假设看到的是k这栋楼:

如果k和i的颜色是一样的,那么会误当成一栋,同时k + 1至i - 1这些楼需要放在k的后面,因为它们都比k矮,那么前一栋看到的就不是k了,那么,k + 1至i - 1这些楼放置的方案数为(假设下标为1),k * (k + 1) * (k + 2) * … * (k + i - 2),若k + 2 等于 i,那么就什么都不用乘,因为i,k之间没有其他楼:f(j,i) += f(j,k) * k * (k + 1) * (k + 2) * … * (k + i - 2);

如果颜色不一样,那么就能多看到一栋楼:f(j,i) += f(j - 1,k) * k * (k + 1) * (k + 2) * … * (k + i - 2)。

时间复杂度:首先n^2种状态,转移O(n),总为O(n^3)。

我们需要再优化一下转移,可以画个表格来寻找方法:



摩天楼1,3,6是同一种颜色,而1,3,5,6中没有一个的颜色和摩天楼7相同,它们的颜色也不一定相同。现在若需要求f(3,7)(蓝色部分),它的值等于:

f(3, 7) = 

+ f(2, 1) * 1 * 2 * 3 * 4 * 5

+ f(3, 2) * 2 * 3 * 4 * 5

f(2, 3) * 3 * 4 * 5

f(3, 4) * 4 * 5

f(2, 5) * 5

+ f(2, 6)

把它分成黄色,绿色,蓝色三个部分求解。

再来看看它前一个同种颜色的摩天楼(第4栋):

f(3, 4) =

+ f(2, 1) * 1 * 2 * 3

+ f(3, 2) * 2 * 3

f(2, 3) * 3  

看到没有,f(3, 7)划线部分的值和f(3, 4)是一样的。并且加粗部分是划线部分的4 * 5倍,黄色部分的值就求出来了!至于i * (i + 1)* … * j,可以预处理。然后,f(3, 4) * 4 * 5可以单独算(绿色部分)。接着,就剩下f(2, 5) * 5 + f(2, 6)。既然第4栋摩天楼是前一个和7同颜色的,那么,4 + 1至7 - 1的楼都不和7相同颜色,这就可以用类似于部分和的东西。 

由于公式编辑麻烦,这里就不贴了,找找规律就知道了。

那么g(2, 6) = 

+ f(2, 1) * 1 * 2 * 3 * 4 * 5

+ f(2, 2) * 2 * 3 * 4 * 5

f(2, 3) * 3 * 4 * 5

f(2, 4) * 4 * 5

f(2, 5) * 5

+ f(2, 6)

g(2, 4) = 

+ f(2, 1) * 1 * 2 * 3 

+ f(2, 2) * 2 * 3 

f(2, 3) * 3 

f(2, 4)

那么我们所需要求的f(2, 5) * 5 + f(2, 6)就等于g(2, 6)减去4 * 5倍的g(2, 4)(橙色部分)。至于g如何求,这应该很简单吧。g(j,i) = g(j,i-1) * i + f(j, i)。

这样,我们就可以O(1)实现求解某个f(j, i)了。问题就到此结束了,时间复杂度就是O(n^2)。


我的代码下标是从0开始的,写的一般般:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;

const ll mod = 1000000009;
const int N = 1300, C = 2707;

char x[N];
int n, L, col[N];	// 第i栋楼的颜色编号 
int pre[C];			// pre[col],前一个颜色为col的摩天楼的标号 
ll f[N][N], 		// 即所说的f数组 
	fac[N][N], 		// fac(i, j) 用来保存 i * (i + 1)* … * j 
	sum[N];			// 即所说的g数组,这里省去了一维 

inline int ord(char a)
{
	if (a >= 'a' && a <= 'z') return (int) a - 'a';
	if (a >= 'A' && a <= 'Z') return (int) a - 'A' + 26;
}

int main()
{
	freopen("sky.in", "r", stdin);
	freopen("sky.out", "w", stdout);
	
	scanf("%d%d\n", &n, &L);
	scanf("%s\n", x);
	for (int i = 0; i < n; i ++)
	{
		char ch;
		scanf("%c", &ch);
		// 题目要求的颜色读入方式是两个字母
		col[i] = ord(x[i]) * (26 << 1) + ord(ch);
	}
	// 加个哨兵,方便输出 
	col[n] = C - 1;
	
	// 把摩天楼的顺序调转 
	for (int i = 0; i < n >> 1; i ++) swap(col[i], col[n - i - 1]);
	
	for (int i = 0; i <= n + 1; i ++)
		for (int j = 0; j <= n + 1; j ++) fac[i][j] = 1LL;
	for (int i = 1; i <= n; i ++)
	{
		fac[i][i] = i;
		for (int j = i + 1; j <= n; j ++)
			fac[i][j] = (fac[i][j - 1] * j) % mod;
	}
	
	for (int j = 1; j <= L + 1; j ++)
	{
		for (int i = 0; i <= n; i ++) pre[col[i]] = -1;
		
		if (j == 1) f[j][0] = 1LL; 
		pre[col[0]] = 0;
		
		sum[0] = f[j - 1][0];
		for (int i = 1; i <= n; i ++)
		{
			if (pre[col[i]] == -1) f[j][i] = sum[i - 1];
			else 
			{
				int p = pre[col[i]];
				if (p > 0)f[j][i] = f[j][p] * fac[p][i - 1] % mod;
				f[j][i] = (f[j][i] + f[j][p] * fac[p + 1][i - 1]) % mod;
				if (p + 1 < i) f[j][i] = (f[j][i] + sum[i - 1] - sum[p] * fac[p + 1][i - 1] % mod + mod) % mod;
			}
			pre[col[i]] = i;
			sum[i] = (sum[i - 1] * i + f[j - 1][i]) % mod;
		}
	}
	
	printf("%I64d\n", f[L + 1][n]);
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值