题目:http://codeforces.com/contest/909/problem/C
题意:给一系列循环(for)和基本语句(simple),求通过for的不同嵌套,能组成多少种不同的程序。
分析:
逆推DP;
f[i][j] 表示:第i-n个字符组成的程序,且有j个独立语句的方案数;
其中独立语句意思是:未被for包含的s语句 或者 未被其他for包含的f语句;
用sum数组记录f[i+1][]的后缀和;
sum[j] = f[i+1][j] + f[i+1][j+1] +..+ f[i+1][n];
那么f[i][j] = f[i+1][j-1] , if s[i]==’s’
= sum[j] , if s[i]==’f’
解释:若s[i]是’s’,则i-n组成j个独立语句需要i+1-n组成j-1个独立语句;
若s[i]是’f’,则这个for可以嵌套好多语句,可以选择嵌套1个,则需要i+1-n组成j个语句,可以选择嵌套2个,则需要i+1-n组成j+1个语句…于是f[i][j]=sum[j].
代码:
#include <bits/stdc++.h>
using namespace std;
const int Tmax=5005;
const long long int MOD=1e9+7;
int n;
char s[Tmax];
long long int f[Tmax][Tmax],ans,sum[Tmax];
void work()
{
int i,j;
sum[1]=1;
for(i=n-1;i>=1;i--)
{
for(j=n-i+1;j>=1;j--)
{
if(s[i]=='s') f[i][j]=f[i+1][j-1];
else f[i][j]=sum[j];
sum[j]=(sum[j+1]+f[i][j])%MOD;
//printf("%d %d:%d\n",i,j,f[i][j]);
}
}
return;
}
int main()
{
int i;
cin>>n;
for(i=1;i<=n;i++) cin>>s[i];
f[n][1]=1;
work();
for(i=1;i<=n;i++)
ans=(ans+f[1][i])%MOD;
cout<<ans;
return 0;
}