CodeForces 360 C.Levko and Strings（组合数学+dp）

Description

给出一个长度为$n$的只由小写字母组成的字符串，定义另一个长度为$n$的只由小写字母组成的字符串$t$与$s$的相关度为二元组$(i,j)$的对数，其中$i,j$需满足$1\le i\le j\le n,t[i,...,j]$字典序大于$s[i,...,j]$，问和$s$相关度为$k$的字符串$t$的个数

Input

第一行两个整数$n,k$表示字符串长度和相关度，之后输入一个长度为$n$，只由小写字母组成的字符串$s$$(1\le n\le 2000,0\le k\le 2000)$

Output

输出和$s$相关度为$k$的字符串个数，结果模$10^9+7$

Sample Input

2 2
yz

Sample Output

26

Solution

以$dp[i][j]$表示$t$的前$i$个字符已经放好，后面的字符和$s$相同且产生了$j$个相关度的方案数

假设$s$和$t$在第$i$个位置之前第一个不同的位置在$l$，即$s[l+1,...,i-1]=t[l+1,...,i-1]$，那么当第$i$个位置$t[i]$取的比$s[i]$大时，起点在$[l+1,i]$之间，终点在$[i,n]$之间的区间都会产生一个相关度，而对于起点在$[1,l]$之间，终点在$[l+1,n]$之间的区间，$s$和$t$不同的位置必然在$l+1$之前，第$i$个位置怎么填不会影响其对相关度的贡献，其相关度的贡献之前已经算过了，故有转移方程$dp[i][j]+=('z'-s[i])\cdot\sum\limits_{l=0}^{i-1} dp[i-l][j-(i-l)\cdot (n-i+1)]$；如果第$i$个位置$t[i]$取的比$s[i]$小时，不会产生新的相关度，故有$dp[i][j]+=(s[i]-'a')\cdot\sum\limits_{l=0}^{i-1}dp[l][j]$，用$sum[j]$维护$\sum\limits_{l=0}^{i-1}dp[l][j]$，则这块$O(1)$转移，最后$\sum\limits_{i=0}^ndp[i][k]$即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=2005;
#define mod 1000000007
int n,k,dp[maxn][maxn],sum[maxn];
char s[maxn];
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int Solve()
{
    sum[0]=dp[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=k;j++)
        {
            for(int l=i-1;l>=0&&(i-l)*(n-i+1)<=j;l--)
                add(dp[i][j],dp[l][j-(i-l)*(n-i+1)]);
            dp[i][j]=(ll)('z'-s[i])*dp[i][j]%mod;
            add(dp[i][j],(ll)(s[i]-'a')*sum[j]%mod);
            add(sum[j],dp[i][j]);
        }
    }
    int ans=0;
    for(int i=0;i<=n;i++)add(ans,dp[i][k]); 
    return ans;
}
int main()
{
    scanf("%d%d",&n,&k);
    scanf("%s",s+1);
    printf("%d\n",Solve());
    return 0;
}