Good Bye 2015 D. New Year and Ancient Prophecy

D. New Year and Ancient Prophecy

time limit per test

2.5 seconds

memory limit per test

512 megabytes

input

standard input

output

standard output

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.

Input

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'.

Output

Print the number of ways to correctly split the given sequence modulo 109 + 7.

Sample test(s)

input

6
123434

output

8

input

8
20152016

output

4

Note

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"

---

 

 

一个长为n的数字字符串，划分成若干子串，保证按照递增顺序的方案数。

f[i][j]表示以i下标为结尾，i所在的子串长度不超过j的方案数。可以得到如下递推式：

f[i][j]=f[i][j-1]， s[i-j]=='0'

         f[i-j][i-j]， i-2*j<0

         f[i-j][j]，   s[i-2*j...i-j-1]<s[i-j...i-1]

         f[i-j][j-1]，s[i-2*j...i-j-1]>=s[i-j...i-1]

那么对于第3、4种情况，需要判定两段字符串的大小，如果从头到尾判断的话，复杂度为O(n)，对于极限数据来说会超时。

解决方法有两种：

1. 将s的所有子串哈希，对于两个起始位置x、y，二分找到最小的不相等的长度，然后比较，复杂度为O(logn)

2. 设low[x][y]为：对于两个起始位置x、y的最小的不相等的长度。则有如下递推式：

low[i][j]= 0，                       s[i-1]!=s[j-1]

               low[i+1][j+1]+1， s[i-1]==s[j-1]

然后直接比较x+low，y+low位，复杂度为O(1)

1. 哈希

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <ctime>
#define oo 1000000007
#define maxn 5200
#define maxm 4200

using namespace std;

int n;
string s;
int f[maxn][maxn];
long long hash[maxn][maxn];
//f[x][y] 表示以x结尾的长度小于等于y的答案

bool judge(int x,int y,int len)
{
    int l=0,r=len-1,d=0;
    while (l<=r)
    {
        int mid=(l+r)>>1;
        if (hash[x][x+mid]!=hash[y][y+mid])
        {
            r=mid-1;
            d=mid;
        }
        else
            l=mid+1;
    }
    return s[x+d-1]<s[y+d-1];
}

int main()
{
    //freopen("D.in","r",stdin);
    scanf("%d",&n);
    cin>>s;
    memset(hash,0,sizeof(hash));
    for (int i=1;i<=n;i++)
        for (int j=i;j<=n;j++)
            hash[i][j]=hash[i][j-1]*131+s[j-1];
    f[0][0]=1;
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=i;j++)
        {
            if (s[i-j]=='0')
            {
                f[i][j]=f[i][j-1];
                continue;
            }
            int p;
            //p是前一段可以取得最长长度
            if (i-2*j>=0)
            {
                if (judge(i-2*j+1,i-j+1,j)) p=j;
                else p=j-1;
            }
            else
                p=i-j;
            f[i][j]=(f[i][j-1]+f[i-j][p])%oo;
        }
    }    
    printf("%d\n",f[n][n]);
    return 0;
}

2.dp

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <ctime>
#define oo 1000000007
#define maxn 5200
#define maxm 4200

using namespace std;

int n;
string s;
int f[maxn][maxn],low[maxn][maxn];
//f[x][y] 表示以x结尾的长度小于等于y的答案
//low[x][y] 表示从x、y开始的最小的不相等的偏移量

bool judge(int x, int y, int len)
{
    int d=low[x][y];
    return d<len && s[x+d-1]<s[y+d-1];
}

int main()
{
    //freopen("D.in","r",stdin);
    scanf("%d",&n);
    cin>>s;
    for (int i=n;i>=1;i--)
        for (int j=n;j>i;j--)
            if (s[i-1]!=s[j-1])
                low[i][j]=0;
            else
                low[i][j]=low[i+1][j+1]+1;
    f[0][0]=1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=i;j++)
        {
            if (s[i-j]=='0')
            {
                f[i][j]=f[i][j-1];
                continue;
            }
            int p;
            //p是前一段可以取得最长长度
            if (i-2*j>=0)
            {
                if (judge(i-2*j+1,i-j+1,j)) p=j;
                else p=j-1;
            }
            else
                p=i-j;
            f[i][j]=(f[i][j-1]+f[i-j][p])%oo;
        }
    printf("%d\n",f[n][n]);
    return 0;
}