codeforces 611D New Year and Ancient Prophecy

f[i = 以i结尾][j = 长度为j] = 方案数。

f[i][j] = sum{ f[i-j][k] , k < j || (k == j && s(i-j+1,j) > s(i-2*j+1,j) ) }

转移为O(N^3)需要优化,

对于k < j,递推g[i][j] = sum(f[i][k], k <= j)。

对于k == j,有O(N^2)个后缀,可以用二维数组lcp[i][j]递推i和j开头的最长公共前缀。

(后缀数组倍增大概也可以做的,用memcpy都可以过的,常数还比较小,极限数据1481ms。

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

typedef long long ll;

const int N = 5e3+5, mod = 1e9+7;
int f[N][N], g[N][N];
char s[N];

int lcp[N][N];

bool bigger(int i, int j, int len)
{
    if(lcp[i][j] >= len) return false;
    else {
        int c = lcp[i][j];
        return s[i+c] > s[j+c];
    }
}

//#define LOCAL
int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif
    int n, i, j;
    scanf("%d%s", &n, s+1);
    if(s[1] == '0'){ puts("0"); return 0; }
    for(i = n; i > 0; i--){
        for(j = n; j > 0; j--){
            if(s[i] == s[j]) lcp[i][j] = lcp[i+1][j+1]+1;
        }
    }

    for(i = 1; i <= n; i++){
        for(j = 1; j < i; j++) {
            g[i][j] = g[i][j-1];
            if(s[i-j+1] != '0'){
                f[i][j] = g[i-j][min(j-1,i-j)];
                if(f[i-j][j] && bigger(i-j+1,i-j-j+1,j)){
                    f[i][j] += f[i-j][j];
                    if(f[i][j] >= mod) f[i][j] -= mod;
                }
                g[i][j] += f[i][j];
                if(g[i][j] >= mod) g[i][j] -= mod;
            }
        }
        f[i][i] = 1;
        g[i][i] = g[i][i-1] + 1;
        if(g[i][i] == mod) g[i][i] = 0;
    }
    printf("%d\n",g[n][n]);
    return 0;
}

 

转载于:https://www.cnblogs.com/jerryRey/p/5091690.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值