CodeForces 909C Python Indentation

Posted 2020-10-19 Fighting Heart

题解：
$dp$。

$dp[i][j]$表示到第$i$个字母，且该字母在第$j$层的方案数，转移分析一下就知道了。

#include <bits/stdc++.h>
using namespace std;

const long long mod = 1e9 + 7;
const int maxn = 	5000 + 10;
int n, f;
char op[maxn][5];
long long dp[maxn][maxn];

int main() {
  scanf("%d", &n);
  for(int i = 1; i <= n; i ++) {
    scanf("%s", op[i]);
  }
  
  dp[1][1] = 1;
  for(int i = 2; i <= n; i ++) {
    if(op[i - 1][0] == \'f\') {
      for(int j = 1; j <= i - 1; j ++) {
        dp[i][j + 1] = dp[i - 1][j];
      }
    } else {
      long long sum = 0;
      for(int j = i - 1; j >= 1; j --) {
        sum = (sum + dp[i - 1][j]) % mod;
        dp[i][j] = sum;
      }
    }
  }
  
  /*
  for(int i = 1; i <= n; i ++) {
    for(int j = 1; j <= i; j ++) {
      printf("%lld ", dp[i][j]);
    }
    printf("\\n");
  }
  */
  
  long long ans = 0;
  if(op[n][0] == \'f\') ans = 0;
  else {
    for(int j = 1; j <= n; j ++) {
      ans = (ans + dp[n][j]) % mod;
    }
    printf("%lld\\n", ans);
  }
  return 0;
}