codeM 美团b轮 E 子串

来源:互联网 发布:北京景点知乎 编辑:程序博客网 时间:2024/05/20 20:57

时间限制: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");}