链接:https://ac.nowcoder.com/acm/problem/13253
来源:牛客网
题目描述
给出一个正整数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
输出
yes
题解
先用暴力求出s,然后用kmp进行字符串匹配。
代码
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef unsigned long long ull;
const ull B=100000007;
string t;
int Next[1000005];
int n;
bool kmp(string a,string b){
int al=a.length(),bl=b.length();
if(al>bl)
return false;
ull t=1;
for(int i=0;i<al;i++)
t*=B;
ull ah=0,bh=0;
for(int i=0;i<al;i++)
ah=ah*B+a[i];
for(int i=0;i<al;i++)
bh=bh*B+b[i];
for(int i=0;i+al<=bl;i++){
if(ah==bh)
return true;
if(i+al<bl)
bh=bh*B+b[i+al]-b[i]*t;
}
return false;
}
char change(int x)
{
if(x>=0&&x<=9)
x=x+'0';
else if(x>=10&&x<=15)
x=x-10+'A';
return (char)x;
}
string numToString(int a,int b)
{
string s;
while(b>0)
{
int tmp=b%a;
s=change(tmp)+s;
b/=a;
}
return s;
}
void makeNext()
{
int j=0,k=-1;
Next[0]=-1;
while(j<t.length())
{
if(k==-1||t[j]==t[k])
{
j++;k++;
Next[j]=k;
}
else
k=Next[k];
}
}
int main()
{
scanf("%d",&n);
cin>>t;
makeNext();
bool flag=false;
for(int i=2;i<=16;i++)
{
string s;
for(int j=1;j<=n;j++)
s+=numToString(i,j);
if(kmp(t,s)==1)
{
flag=true;
break;
}
}
if(flag)
printf("yes\n");
else
printf("no\n");
return 0;
}