Codeforces Round #455 (Div. 2)C_Python Indentation_DP

题意

python语言有两种语句，simple语句（s语句）和for语句（f语句）。其中for语句的循环体需要缩进一格，并且不能为空（for语句的循环体里至少有一个s语句或for语句）。给出n条语句，问总共有多少种合法的缩进方法。

思路

dp[i][j] 第i行缩进j格的种数。
如果第i行是f语句，它的下一行只能放在它的循环体里，缩进格数只能加一。dp[i+1][j+1] = dp[i][j]
如果第i行是s语句，它的下一行可以放在之前任意一个for 的循环体中，即下一行的缩进格数小于等于这一行就合法。dp[i+1][j] = sigma(dp[i][k]) (j <= k <= n)

链接

http://codeforces.com/contest/909/problem/C

代码

#include<bits/stdc++.h>

using namespace std;

const int maxn = 5000 + 10;
const int mod = 1e9 + 7;

int n;
char c;
int dp[maxn][maxn];

int main(){
    while(scanf("%d", &n) == 1){
        memset(dp, 0, sizeof dp);
        dp[1][0] = 1;

        for(int i = 1; i <= n; ++i){
            scanf(" %c", &c);

            if(c == 's'){
                int sum = 0;
                for(int j = n; j >= 0; --j){
                    sum = (sum + dp[i][j]) % mod;
                    dp[i+1][j] = sum;
                }
            }else{
                for(int j = 0; j <= n; ++j){
                    dp[i+1][j+1] = dp[i][j];
                }
            }

        }

        int res = 0;
        for(int i = 0; i <= n; ++i){
            res = (res + dp[n][i]) % mod;
        }
        cout << res << endl;
    }

    return 0;
}