美团CODEM 字串

时间限制：1秒
空间限制：32768K
给出一个正整数n，我们把1..n在k进制下的表示连起来记为s(n,k)，例如s(16,16)=123456789ABCDEF10, s(5,2)=11011100101。现在对于给定的n和字符串t，我们想知道是否存在一个k(2 ≤ k ≤ 16)，使得t是s(n,k)的子串。
输入描述:
第一行一个整数n(1 ≤ n ≤ 50,000)。
第二行一个字符串t(长度 ≤ 1,000,000)
输出描述:
“yes”表示存在满足条件的k，否则输出”no”
输入例子:
8
01112
输出例子:
yes

解法：直接暴力拆分然后做一个KMP，但是要注意用下标为1的KMP在串是空串的时候要GG.所以选择下标为0的KMP的写法。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3000010;
char s1[maxn], s2[maxn];
string s;
int fail[maxn];
int n,len1,len2;
int KMP()
{
    int i=0, j=0;
    while(i<len1){
        if(j==-1||s1[i]==s2[j]){
            i++;
            j++;
        }
        else{
            j=fail[j];
        }
        if(j==len2) return 1;
    }
    return 0;
}
int main()
{
    scanf("%d",&n);
    scanf("%s", s2);
    len2 = strlen(s2);
    int i=0, j=-1;
    fail[0]=-1;
    while(i<len2){
        if(j==-1||s2[i]==s2[j]){
            i++;
            j++;
            fail[i]=j;
        }
        else j=fail[j];
    }
    int ans = 0;
    for(int jz = 2; jz<=16; jz++){
        len1 = 0;
        for(int i=1; i<=n; i++){
            int x = i, cnt = 0, num[100];
            while(x){
                num[cnt++] = x%jz;
                x/=jz;
            }
            for(int j=cnt-1; j>=0; j--){
                if(num[j]<10){
                    s1[len1++]=num[j]+'0';
                }
                else{
                    s1[len1++]=num[j]-10+'A';
                }
            }
        }
        ans = max(ans, KMP());
        if(ans > 0) break;
    }
    if(ans > 0) puts("yes");
    else puts("no");
    return 0;
}