KMP -牛客 NC13253 子串

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

分析：

先按照题意把1-n的数字转化为2-16进制的字符串，再用kmp来判断是否t出现在字符串中。或者用strstr（）函数也可以来判断。
注意：转化k进制的时候，大于9的数字转化为大写字母A-F，并且转化的时候需要倒叙存入目标字符串。

kmp

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e6+10;
char a[N],b[N],c[N];
int ne[N];
void get_ne()
{
	int l=strlen(a);
	int i=0,j=-1;
	ne[0]=-1;
	while(i<l)
	{
		if(j==-1||a[i]==a[j])
		{
			i++,j++;
			if(a[i]!=a[j])
			ne[i]=j;
			else ne[i]=ne[j];
		}
		else j=ne[j];
	}
}
int kmp(int n)
{
	int l=strlen(a);
	int i=0,j=0;
	while(i<n&&j<l)
	{
		if(j==-1||b[i]==a[j]) i++,j++;
		else j=ne[j];
		
	}
	if(j==l) return 1;
	return 0;
}
int main()
{
	int n;
	scanf("%d",&n);
	//getchar();
	scanf("%s",a);
	int l=strlen(a);
	int mx=2,x,k,f=0,y,t;
	get_ne();
	for(int i=2;i<=16;i++)
	{
		k=-1;
		for(int j=1;j<=n;j++)
		{
			t=-1;
			x=j;
			while(x)
			{
				y=x%i; 
				if(y>9)
				{
					c[++t]=y-10+'A'; 
				}
				else c[++t]=y+'0';
				x=x/i;
			}
			for(int z=t;z>=0;z--)
			{
				b[++k]=c[z];
			}
		}
		b[++k]='\0';
		if(kmp(k))
		{
			f=1;
			break;
		}
	}
	if(f==1) printf("yes");
	else printf("no");
	return 0;
 } 

strstr（）函数

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e6+10;
char a[N],b[N],c[N];
int main()
{
	int n;
	scanf("%d",&n);
	//getchar();
	scanf("%s",a);
	int l=strlen(a);
	int mx=2,x,k,f=0,y,t;
	for(int i=2;i<=16;i++)
	{
		k=-1;
		for(int j=1;j<=n;j++)
		{
			t=-1;
			x=j;
			while(x)
			{
				y=x%i; 
				if(y>9)
				{
					c[++t]=y-10+'A'; 
				}
				else c[++t]=y+'0';
				x=x/i;
			}
			for(int z=t;z>=0;z--)
			{
				b[++k]=c[z];
			}
		}
		b[++k]='\0';
		if(strstr(b,a)!=NULL)
		{
			f=1;
			break;
		}
	}
	if(f==1) printf("yes");
	else printf("no");
	return 0;
 }