codeM 美团b轮 E 子串

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

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
string mo;
int f[N+100];
void getfill(string s)  
{  
    memset(f,0,sizeof(f));  
    for(int i=1;i<s.size();i++)  
    {  
        int j=f[i];  
        while(j && s[i]!=s[j])  
            j=f[j];  
        f[i+1]=(s[i]==s[j])?j+1:0;  
    }  
} 

int find(string a)  
{  

    int ans=0;  
    int j=0;  
    for(int i=0;i<a.size();i++)  
    {  
        while(j && a[i]!=mo[j])  
            j=f[j];  
        if(a[i]==mo[j])  
            j++;  
        if(j==mo.size()){  
            return 1; 
        }  
    }  
    return 0;  
} 

char ccc(int k)
{
    if(k==10)
    {
        return 'A';
    }
        if(k==11)
    {
        return 'B';
    }
    if(k==12)
    {
        return 'C';
    }
        if(k==13)
    {
        return 'D';
    }
        if(k==14)
    {
        return 'E';
    }
        if(k==15)
    {
        return 'F';
    }
    return k+'0';
}


int main()
{
    int n;
    cin>>n;
    cin>>mo;
    getfill(mo);
    string g;

    for(int i=2;i<=16;i++)
    {
        g.clear();
        for(int j=1;j<=n;j++)
        {
            string t;
            int k=j;
            while(k)
            {
                int h=k%i;
                k/=i;
                t+=ccc(h);
            }
            reverse(t.begin(),t.end());
            g+=t;
        }
        if(find(g))
        {
            printf("yes\n");
            return 0;
        }
    }
    printf("no\n");
}