Codeforces1029A Many Equal Substrings (KMP找循环节）

题目链接

http://codeforces.com/problemset/problem/1029/A

题意

给定一个长度为n的由小写字母组成的字符串t，要求构造一个字符串s。
s的长度尽可能的短。
s包含k个子串t。

题解

观察第一个样例。
3 4
aba
找到串t的最小循环节ab。

当前字符串t已经有一个t了，只需再添加若干个字符再凑出k-1个t。

根据kmp算法找失配指针pre。
找到aba最后一个字符的失配位置pre[3]=1.
然后在t后面添加k-1个t[2..n].即ba。
最后得到aba ba ba ba

AC代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=100;

int pre[maxn],n,k;
char s[maxn];
void getPre()
{
    memset(pre,0,sizeof(pre));

    int j=0;
    for(int i=2;i<=n;i++)
    {
        while(j>0 && s[j+1]!=s[i]) j=pre[j];
        if(s[j+1]==s[i]) j++;
        pre[i]=j;
    }
}

int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        scanf("%s",s+1);
        getPre();
        //for(int i=1;i<=n;i++)
          //  printf("%d %d\n",i,pre[i]);
        printf("%s",s+1);
        for(int i=1;i<=k-1;i++)
        {
            for(int j=pre[n]+1;j<=n;j++)
                printf("%c",s[j]);
        }
        putchar(10);
    }
    return 0;
}