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;
}