CodeM初赛B轮

做什么啊，我这么菜，应该弃赛的

[编程|1500分] 子串

时间限制：3秒
空间限制：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就好了，我想到了KMP但是没有从网上找模板，找到模板发现自己还是想的不好，最后优化成了这个样子，自我感觉还是很完美的

长度 ≤ 1,000,000，这个也是给我开数组的提示，竟然不坑，反正觉得2进制的1到50k还是不小的

#include <bits/stdc++.h>
using namespace std;
char s[],t[];
int nex[];
int l;
char f[];
void la(int n, int b) {
    static char c[]= {'','','','','','','','','','','A','B','C','D','E','F'};
    l=;
    while(n) {
        f[l++]=c[n%b];
        n=n/b;
    }
}
void pre(char *p) {
    int i,n,k;
    n=strlen(p);
    nex[]=nex[]=;
    k=;
    for (i=; i<=n; i++) {
        for(; k!=&&p[k]!= p[i-]; k=nex[k]);
        if(p[k]==p[i-])k++;
        nex[i]=k;
    }
}
int kmp(char *text, char *p) {
    int m,n,s,q;
    m=strlen(p);
    n=strlen(text);
    q=s=;
    while (s<n) {
        for (q=nex[q]; q<m&&p[q]==text[s]; q++,s++);
        if (q== ) s++;
        else if(q==m) {
            return ;
        }
    }
    return ;
}
int main() {
    int n;
    scanf("%d",&n);
    getchar();
    for(int i=; ; i++) {
        char c=getchar();
        if(c=='\n') {
            t[i]=;
            break;
        }
        t[i]=c;
    }
    pre(t);
    for(int j=; j>; j--) {
        int b=;
        for(int i=; i<=n; i++) {
            la(i,j);
            for(int k=l-; k>=; k--)
                s[b++]=f[k];
        }
        s[b]=;
        if(kmp(s,t))
            return *puts("yes");
    }
    return *puts("no");
}