gmoj.5219 【GDOI2018模拟7.10】B

题目

题目描述

对于一个排列，考虑相邻的两个元素，如果后面一个比前面一个大，表示这个位置是上升的，
用 I 表示，反之这个位置是下降的，用 D 表示。如排列 3,1,2,7,4,6,5 可以表示为 DIIDID。
现在给出一个长度为 n-1 的排列表示，问有多少种 1 到 n 的排列满足这种表示。

输入

一个字符串 S，S 由 I，D，？组成。？表示这个位置既可以为 I，又可以为 D。

输出

有多少种排列满足上述字符串。输出排列数模 1000000007。

样例输入

?D

样例输出

3

数据范围

对于 20%的数据，S 长度 ≤ 10；
对于 100%的数据，S 长度 ≤ 1000。

---

题解

题外话：这道题目和 UVa 1650 重题了。
这道题目一眼看上去就是一道DP，吓得我比赛时直接弃疗打暴力。
20分暴力太水了，这里就不讲了。
以下是正解：
设$f_{i,j}$表示到了第 i 位，填写的数字为 j 的方案数。
但是这样显然无法转移。
不难发现，把它改成这个样子就可以转移了：

设$f_{i,j}$表示到了第 i 位，填写的数字为 j 的方案数，且填写的数字都在[1…i]中。

下面就来证明这两种状态是等价的。
先来看一下题目给出的一个数列{3,1,2,7,4,6,5}，它们是满足字符串DIIDID的。我们不妨把大于3的数字全部+1，数列就变成了{4,1,2,8,4,7,6}。可以发现它此时还是满足DIIDID的。
经过多次试验，这个结论仍然成立，因此可以得出结论：
{1,2,…,j,j+1,…,i}等价于{1,2,…,j-1,j+1,…i}，即[1…i]中任选i-1个数的数列等价于[1…i-1]排列的数列。以此类推。

就这样，我们把状态定了下来，接下来就来列状态转移方程了。

当s='I’时

上一个数小于第 i 位的数 j 时，上一个数就一定在区间[1…j-1]中，因此可以列出方程$$f_{i,j}=\sum _{k=1}^{j-1}{f}_{i-1,k}$$

当s='D’时

上一个数大于第 i 位的数 j 时，上一个数就一定在区间[j+1…i]中。状态转移方程显然是：
$$f_{i,j}=\sum _{k=j+1}^i{f}_{i-1,k}$$
但由于我们DP数组中没有$f_{i-1,i}$的状态，而它们又是合法的，所以我们要想办法处理这个状态。
根据上文的推导得出的结论

{1,2,…,j,j+1,…,i}等价于{1,2,…,j-1,j+1,…i}，即[1…i]中任选i-1个数的数列等价于[1…i-1]排列的数列。

所以这些以 j+1～i 结尾的状态也等价于以 j～i-1 结尾的状态。状态转移方程可以表示为
$$f_{i,j}=\sum _{k=j}^{i-1}{f}_{i-1,k}$$

当s=’?'时

这也就是把上两种情况结合起来，这里就不讲了。

---

温馨提示

如果按照状态转移方程来敲代码，显然会时超，因此要用前缀和优化。

---

代码

#include<cstdio>
#include<cstring>
using namespace std;
#define mod 1000000007
#define N 1005
int f[N][N],sum[N][N];
char s[N];
int main()
{
	int n,i,j;
	scanf("%s",s+1);
	n=strlen(s+1)+1;
	sum[1][1]=1;
	for(i=2;i<=n;i++)
	{
		for(j=1;j<=i;j++)
		{
			if(s[i-1]=='I'||s[i-1]=='?') f[i][j]=sum[i-1][j-1];
			if(s[i-1]=='D'||s[i-1]=='?')
			{
				f[i][j]=(f[i][j]+sum[i-1][i-1]-sum[i-1][j-1])%mod;
				if(f[i][j]<0) f[i][j]+=mod;
			}
			sum[i][j]=(sum[i][j-1]+f[i][j])%mod;
		}
	}
	printf("%d\n",sum[n][n]);
	return 0;
}