Good Bye 2015 D. New Year and Ancient Prophecy
Limak is a little polar bear. In the snow he found a scroll with the ancient prophecy. Limak doesn't know any ancient languages and thus is unable to understand the prophecy. But he knows digits!
One fragment of the prophecy is a sequence of n digits. The first digit isn't zero. Limak thinks that it's a list of some special years. It's hard to see any commas or spaces, so maybe ancient people didn't use them. Now Limak wonders what years are listed there.
Limak assumes three things:
- Years are listed in the strictly increasing order;
- Every year is a positive integer number;
- There are no leading zeros.
Limak is going to consider all possible ways to split a sequence into numbers (years), satisfying the conditions above. He will do it without any help. However, he asked you to tell him the number of ways to do so. Since this number may be very large, you are only asked to calculate it modulo 109 + 7.
The first line of the input contains a single integer n (1 ≤ n ≤ 5000) — the number of digits.
The second line contains a string of digits and has length equal to n. It's guaranteed that the first digit is not '0'.
Print the number of ways to correctly split the given sequence modulo 109 + 7.
6 123434
8
8 20152016
4
In the first sample there are 8 ways to split the sequence:
- "123434" = "123434" (maybe the given sequence is just one big number)
- "123434" = "1" + "23434"
- "123434" = "12" + "3434"
- "123434" = "123" + "434"
- "123434" = "1" + "23" + "434"
- "123434" = "1" + "2" + "3434"
- "123434" = "1" + "2" + "3" + "434"
- "123434" = "1" + "2" + "3" + "4" + "34"
Note that we don't count a split "123434" = "12" + "34" + "34" because numbers have to be strictly increasing.
In the second sample there are 4 ways:
- "20152016" = "20152016"
- "20152016" = "20" + "152016"
- "20152016" = "201" + "52016"
- "20152016" = "2015" + "2016"
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;
typedef __int64 ll;
#define inf 99999999
#define pi acos(-1.0)
#define MOD 1000000007
#define maxn 5050
ll dp[maxn][maxn];
int nxt[maxn][maxn];
ll c[maxn][maxn];
char s[maxn];
int panduan(int x,int y,int xx,int yy)
{
int i,j,p;
p=nxt[x][xx];
if(p<=y-x){
if(s[x+p]>=s[xx+p]){
return 1;
}
return 0;
}
else return 1;
}
int main()
{
int n,m,i,j,l,r,p;
while(scanf("%d",&n)!=EOF)
{
scanf("%s",s+1);
//memset(dp,0,sizeof(dp));
//memset(nxt,0,sizeof(nxt));
for(i=1;i<=n;i++){
if(s[i]==s[n]){
nxt[i][n]=nxt[n][i]=1;
}
else nxt[i][n]=nxt[n][i]=0;
}
for(i=n-1;i>=1;i--){
for(j=n-1;j>=1;j--){
if(s[i]!=s[j]){
nxt[i][j]=0;
}
else{
nxt[i][j]=nxt[i+1][j+1]+1;
}
}
}
for(j=1;j<=n;j++){
for(i=1;i<=j;i++){
if(s[i]=='0')continue;
if(i==1){
dp[i][j]=1;continue;
}
p=2*i-j-1;
if(p<1){
p=1;
}
else{
if(panduan(p,i-1,i,j)){
p++;
}
}
if(p>=1 && p<=i-1){
dp[i][j]=c[p][i-1]%MOD;
}
}
c[j][j]=dp[j][j]%MOD;
for(i=j-1;i>=1;i--){
c[i][j]=(c[i+1][j]+dp[i][j])%MOD;
}
}
ll ans=0;
for(i=1;i<=n;i++){
ans=(ans+dp[i][n])%MOD;
}
printf("%I64d\n",ans);
}
return 0;
}