美团codeM预赛B 子串

子串

时间限制：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"

输入例子1:

8
01112

输出例子1:

yes

解题思路：暴力枚举2~16形成的字符串，然后进行匹配，匹配可以用kmp来

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <stack>
#include <cmath>
#include <map>
#include <bitset>
#include <set>
#include <vector>
#include <functional>

using namespace std;

#define LL long long
const int INF = 0x3f3f3f3f;

char s1[1000009],s2[1000009],s3[1000009];
int n,len1,len2,nt[1000009];

void getnt()
{
    nt[0]=-1;
    int len=strlen(s2);
    for(int i=0;i<len;i++)
    {
        int k=nt[i];
        while(k>=0&&s2[i]!=s2[k]) k=nt[k];
        nt[i+1]=k+1;
    }
}

int check()
{
    int i=0,j=0;
    while(i<len1)
    {
        if(s1[i]==s2[j]||j==-1) i++,j++;
        else j=nt[j];
        if(j>=len2) return 1;
    }
    return 0;
}

int main()
{
    while(~scanf("%d%s",&n,s2))
    {
        getnt();
        len2=strlen(s2);
        int flag=0;
        for(int i=2;i<=16;i++)
        {
            len1=0;
            for(int j=1;j<=n;j++)
            {
                int k=j,len3=0;
                while(k)
                {
                    if(k%i>=10) s3[len3++]=k%i-10+'A';
                    else s3[len3++]=k%i+'0';
                    k/=i;
                }
                for(int p=len3-1;p>=0;p--) s1[len1++]=s3[p];
            }
            if(check()) flag=1;
            if(flag) break;
        }
        if(flag) printf("yes\n");
        else printf("no\n");
    }
    return 0;
}