Codeforces611 D.New Year and Ancient Prophecy（dp+lcp）

题意：

答案对1e9+7取模

数据范围：n<=5e3

解法：

d[i][j]表示当前长度为i,这一段的左端点为j的方案数.
转移方程:d[i][j]=sum(d[j-1][k]),其中s[k,j-1]<s[j,i].
枚举i,枚举j,枚举k,复杂度是O(n^3)的,
考虑到k的取值有限,当s[k,j-1]的长度<s[j,i]的长度时,
可以直接d[i][j]+=d[j-1][k],因为一定合法,
因为这一段k的取值是连续的,因此可以用前缀和优化,
当s[k,j-1]的长度=s[j,i]时候,需要判断大小关系,
其实就是找第一位不同的数比大小,找第一位不同的数可以用lcp优化.
求lcp可以O(n^2)预处理.

code：

#include <bits/stdc++.h>
using namespace std;
const int maxm=5e3+5;
const int mod=1e9+7;
int lcp[maxm][maxm];
int sum[maxm][maxm];
int d[maxm][maxm];
char s[maxm];
int n;
bool check(int lc,int j){
    if(lcp[lc][j]>=j-lc)return 0;//lcp超过区间长度,那么相等
    char x=s[j+lcp[lc][j]];
    char y=s[lc+lcp[lc][j]];
    return x>y;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    cin>>n;
    cin>>(s+1);
    for(int i=n;i>=1;i--){//预处理lcp
        for(int j=n;j>=1;j--){
            if(s[i]==s[j]){
                lcp[i][j]=lcp[i+1][j+1]+1;
            }
        }
    }
    for(int i=1;i<=n;i++){
        d[i][1]=1;
        for(int j=2;j<=i;j++){
            if(s[j]=='0')continue;//前导零
            int len=i-j+1;
            int lc=max(j-len+1,1);
            d[i][j]=(d[i][j]+sum[j-1][j-1]-sum[j-1][lc-1]+mod)%mod;
            lc--;
            if(lc>=1&&check(lc,j)){
                d[i][j]=(d[i][j]+d[j-1][lc])%mod;
            }
        }
        for(int j=1;j<=i;j++){//维护一个前缀和
            sum[i][j]=(sum[i][j-1]+d[i][j])%mod;
        }
    }
    int ans=0;
    for(int j=1;j<=n;j++){
        ans=(ans+d[n][j])%mod;
    }
    cout<<ans<<endl;
    return 0;
}

---