KMP - 模板(循环节个数) - Period-HDU - 1358+Power Strings-POJ - 2406

KMP - 模板(循环节个数) - Period-HDU - 1358+Power Strings-POJ - 2406

一：Period-HDU - 1358

题目：

一个字符串的前缀是从第一个字符开始的连续若干个字符，例如”abaab”共有5个前缀，分别是a, ab, aba, abaa, abaab。

我们希望知道一个N位字符串S的前缀是否具有循环节。

换言之，对于每一个从头开始的长度为 i （i>1）的前缀，是否由重复出现的子串A组成，即 AAA…A （A重复出现K次,K>1）。

如果存在，请找出最短的循环节对应的K值（也就是这个前缀串的所有可能重复节中，最大的K值）。

输入格式
输入包括多组测试数据，每组测试数据包括两行。

第一行输入字符串S的长度N。

第二行输入字符串S。

输入数据以只包括一个0的行作为结尾。

输出格式
对于每组测试数据，第一行输出 “Test case #” 和测试数据的编号。

接下来的每一行，输出具有循环节的前缀的长度i和其对应K，中间用一个空格隔开。

前缀长度需要升序排列。

在每组测试数据的最后输出一个空行。

数据范围
2≤N≤1000000
输入样例：
3
aaa
4
abcd
12
aabaabaabaab
0
输出样例：
Test case #1
2 2
3 3

Test case #2

Test case #3
2 2
6 2
9 3
12 4

题意：

$求给定长度为n的字符串的n个前缀中，长度为i的“循环节”，以及“循环节”的个数n[i]，1<=i<=\frac{n}{2}。$

$$\begin{gathered} 样例：aabaabaabaab \\ 长度为2的前缀"aa"的循环节为a，有2个。 \\ 长度为6的前缀"aabaab"的循环节为aab，有2个。 \\ 长度为9的前缀"aabaabaab"的循环节为aab，有3个。 \\ 长度为12的前缀"aabaabaabaab"的循环节为aab，有4个。 \end{gathered}$$

题解：

$首要问题是如何找出循环节。$

$通过KMP算法：$

$KMP算法中：数组Next[j]的含义指前j-1个字符构成的子串的最长相同前后缀的长度，设Next[j]=l。$

$那么可知红色部分前缀与蓝色部分后缀长度相等，都为l。从而可知绿色圈圈部分的两条线段长度相等。$

$下面一根线段是红色部分前缀，将其平移，每一段均与蓝色后缀部分对应相等，$

$$\begin{gathered} 即2=5，3=6，4=7，由1=5知1=2，由2=6知2=3，由3=7知3=4，所以1=2=3=4， \\ 1，2，3，4即该字符串的循环节。 \end{gathered}$$

$$\begin{gathered} 于是前j-1个子串的循环节的最小长度就是j-Next[j]，这必然是最小的循环节， \\ 这是因为Next[j]数组存储的是相同前后缀的最大长度。 \end{gathered}$$

$循环节的个数即：子串长度i/循环节长度t。$

$注意：判定是否为循环节还需判断t能否整除i，若不能，则说明不能形成循环节。$

---

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N=1e6+10;
int Next[N];
char str[N];
int n;

void get_next()
{
    for(int i=2,j=0;i<=n;i++)
    {
        while(j&&str[i]!=str[j+1]) j=Next[j];
        if(str[i]==str[j+1]) j++;
        Next[i]=j;
    }
}
int main()
{
    int T=1;
    while(~scanf("%d",&n),n)
    {
        scanf("%s",str+1);
        
        get_next();

        printf("Test case #%d\n",T++);
        for(int i=2;i<=n;i++)
        {
            int t = i - Next[i];
            if (i > t && i % t == 0) printf("%d %d\n", i, i / t);   ///判断条件
        }
        puts("");
    }

    return 0;
}

下标从0开始的Next数组求法：

$写法①、下标从0开始的next数组对应为下标从1开始的next数组的值-1。$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N=1e6+10;
int ne[N];
char str[N];
int n;

void get_next()
{
    ne[0]=-1;
    for(int i=1,j=-1;i<n;i++)
    {
        while(j>=0&&str[i]!=str[j+1]) j=ne[j];
        if(str[i]==str[j+1]) j++;
        ne[i]=j;
    }
}

int main()
{
    int T=1;
    while(~scanf("%d",&n),n)
    {
        scanf("%s",str);

        get_next();

        printf("Test case #%d\n",T++);
        for(int i=1;i<n;i++)
        {
            int t = i + 1 - (ne[i]+1);   //i+1是前缀长度,ne[i]+1是最大共同前后缀的长度
            if ((i+1)/t > 1 && (i+1) % t == 0) printf("%d %d\n", i+1, (i+1) / t);
        }
        puts("");
    }

    return 0;
}

$写法②、对应的next数组的下标仍然从1开始$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N=1e6+10;
int ne[N];
char str[N];
int n;

void get_next(char *str)
{
    int i=0,j=-1,n=strlen(str);
    ne[i]=j;
    while(i<n){
        if(j==-1 || str[i]==str[j]) ne[++i]=++j;
        else j=ne[j];
    }
}

int main()
{
    int T=1;
    while(~scanf("%d",&n),n)
    {
        scanf("%s",str);

        get_next(str);

        printf("Test case #%d\n",T++);
        for(int i=2;i<=n;i++)
        {
            int t = i - ne[i];
            if (i % t == 0 && i/t > 1) printf("%d %d\n", i, i/ t);   ///判断条件
        }
        puts("");
    }

    return 0;
}

---

二：Power Strings-POJ - 2406

题目：
假设s可以由t重复k次拼成，即s=tttt……tt，我们称为s=t^k。先给定一个字符串s，求最大的n使得存在t满足s=tⁿ。

Input
多组数据，每行一个字符串(仅包含可打印字符且长度不超过1000000)，以单独一行.作为终结标志

Output
每组数据一行答案

Sample Input
abcd
aaaa
ababab
.

Sample Output
1
4
3

题解：

$如上题，唯一的区别在于母串无循环节的情况下，整个字符串视作一个“循环节”，输出“1”即可。$

---

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e6+10;
int m;
char p[N];
int Next[N];

void get_next()
{
    for(int i=2,j=0;i<=m;i++)
    {
        while(j&&p[i]!=p[j+1]) j=Next[j];
        if(p[i]==p[j+1]) j++;
        Next[i]=j;
    }
}

int main()
{
    while(scanf("%s",p+1),p[1]!='.')
    {
        m=strlen(p+1);

        get_next();

        int len=m-Next[m];
        if(m%len==0) printf("%d\n",m/len);
        else printf("1\n");
    }

    return 0;
}