【牛客】13253 子串

链接：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;
}