Zoj 3543 Number String (dp) - 2011 ACM-ICPC Dalian Regional Contest Problem E

又是一道dp。比赛时以为是数学题，一直在找规律推公式。

/**

题意：
由{1,2,3}组成的一个排列132，对应一个字符串"ID"，'I'表示Increase，'D'表示Decrease，
对于排列"132"，因为 1 < 3 > 2，所以对应的字符串为"ID"。现在反过来，输入一个字符串，
仅包含'I','D','?'三种字符，字符串长度n(n<=1000)，输出由{1...n+1}组成的排列中，满足
该字符串的个数与1000000007求余。
题解：
dp[i][j] 表示长度为i以j结尾的合法的排列个数（由1...i组成的排列）。
还要清楚一个结论：
给定一个长度为i-1的字符串，由{1,2,...,i}组成的合法排列数和由{1,2,...,j-1,j+1,...,i+1}
组成的合法排列数是相同的。
if ( s[i] == 'I' ) dp[i][j] = dp[i-1][1] + dp[i-1][2] + ... + dp[i-1][j-1];
if ( s[i] == 'D' ) dp[i][j] = dp[i-1][j] + dp[i-1][j+1] +...+ dp[i-1][i-1];
if ( s[i] == '?' ) dp[i][j] = dp[i-1][1] + dp[i-1][2] + ... + dp[i-1][i-1];


时间复杂度是O(n^3)的，但定义一个sum[i][j]利用部分求和，就转化成了O(n^2).

**/

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

typedef long long LL;
const int MAXN = 1000 + 5;
const LL  MOD  = 1000000007;

char s[MAXN];
LL   dp[MAXN][MAXN], sum[MAXN][MAXN];

int main()
{
    while ( scanf("%s", s+2) != EOF )
    {
        int len = strlen(s+2) + 1;
        sum[1][1] = dp[1][1] = 1;
        for (int i = 2; i <= len; i++)
        {
            for (int j = 1; j <= i; j++)
            {
                if ( s[i] == 'I' )
                    dp[i][j] = sum[i-1][j-1];
                else if ( s[i] == 'D' )
                    dp[i][j] = sum[i-1][i-1] - sum[i-1][j-1];
                else
                    dp[i][j] = sum[i-1][i-1];
                sum[i][j] = (sum[i][j-1] + dp[i][j]) % MOD;
            }
        }
        printf("%lld\n", (sum[len][len] + MOD) % MOD); // 因为sum中都保存的是求余之后的值，前面的想减可能产生负数
    }
    return 0;
}