参考资料:学算法的过程很重要
bilibili两个视频:
KMP算法之求next数组代码讲解_哔哩哔哩_bilibili
kmp算法的基本思想,过程(子串中的第i个与主串当前位置进行比较),减少了指针的回溯
求next数组的推导过程,通过图解递归的思想在里面
看后问题:
1.next数组用来干嘛?
2.如何写出kmp的过程?
next 是下一步数组主要记录子串中的第i个与主串当前位置进行比较
kmp函数则需要模拟整个过程:
从第一个数组开始数,当遇到不匹配时,查看子串相同位置的next数值,则转移到next值的位置进行比较,直到next为0时再也无法匹配,如果next不为零则继续匹配,直到s2指针的位置到达子串的长度为止。
#include<bits/stdc++.h>
const int N=1e6+5;
int num;
using namespace std;
string str1,str2;//str1为总的字符串 str2为匹配的字符串
int next1[N];//next[i]代表str2字符串i位置之前的最长前缀
void getnext(){
next1[1]=0;
int i=1,j=0;
while(i<=str2.length()){
if(j==0||str2[i]==str2[j]) next1[++i]=++j;
else j=next1[j];
}
}
int kmp(string str1,string str2)//kmp算法,如果匹配成功,返回1,否则,返回0
{
int s1=0,s2=0;//这是两个指针,s1刚开始指向str1字符串的第一个字符,s2刚开始指向str2字符串的第一个字符
while (s1<str1.size()&&s2<str2.size())//两个指针都没有到达最后一个字符时,执行下面过程
{
if(str1[s1]==str2[s2])//str1字符串s1位置上的字符和str2字符串s2位置上的字符相等
{
s1++;//让两个指针都往后移动一位
s2++;
}
else if(next1[s2]==0)//如果程序运行到这里,说明上面的条件不满足,str1字符串s1位置上的字符和str2字符串s2位置上的字符不相等,如果这个条件满足,说明str1字符串第一个字符就不和str2字符串匹配,让s1往后移动一个位置
s1++;j
else
s2=next1[s2];//这是语句是kmp算法的核心内容,也是kmp算法为什么快的原因
}
if(s2==str2.size())//这个条件如果满足,说明str2字符串都已经匹配完了,并且匹配成功了,返回1
return 1;
else
return 0;
}
string change(int n,int k)
{
string s="";
while(n)
{
if(n%k>=10){
s=char(n%k-10+'A')+s;
}
else s=char(n%k+'0')+s;
n=n/k;;
}
return s;
}
int main(){
ios::sync_with_stdio(false);
cin>>num>>str2;
getnext();
int flag=0;
for(int i=2;i<=16;i++){
str1="";
for(int j=1;j<=num;j++){
str1+=change(j,i);
}
if(kmp(str1,str2)){
flag=1;
break;
}
}
if(flag) puts("yes");
else puts("no");
return 0;
}