题面见传送门
题目大意:给定字符串s,一个整数k,创建一个字符串使得该字符串恰好有k个字串等于s,并且尽可能使创建的字串的长度最短。
分析:
分为两种情况:
1.给定的s的所有前缀,后缀中没有相等的, 如abc, cat;
2.s中有相等的前后缀,如aba,ababa
KMP算法中next数组中next[i]存的就是前i-1个字符组成的字符串中最长相等前后缀的长度,如ababa的next数组为:-1,0, 0, 1, 2,其中next[4]的值为2,前4个字符组成的字符相等的前后缀中长度最长的是ab(整个字符串当前后缀的情况不考虑)。
而输出的答案就是在输出s的基础上,把与前缀相等的后缀中长度最长的串输出k-1遍,如样例
3 2
aba
输出的答案先输出aba,然后把ba输出1遍,也就是ababa。
因为题目中说明了s中只有小写字母,因此为了获取s相等的前后缀中的最长长度,把s[n]设置为’A’,那么next[n]的值就是s的相等的前后缀中长度最长的后缀的长度。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
void Get_next(char s[], int next[])
{
int i = 0, j = -1;
next[0] = -1;
int len = strlen(s);
while(i < len)
{
if(j == -1 || s[i] == s[j])
{
next[++i] = ++j;
}
else
{
j = next[j];
}
}
}
int main()
{
int n, k, next[55];
char s[55];
while(~scanf("%d%d", &n, &k))
{
memset(next, 0, sizeof(next));
scanf("%s", s);
s[n] = 'A';
Get_next(s, next);
// for(int i=0; i<=n; i++)
// printf("%d ", next[i]);
for(int i = 0; i < n; i++)
{
printf("%c", s[i]);
}
k--;
while(k--)
{
for(int i = next[n]; i < n; i++)
{
printf("%c", s[i]);
}
}
printf("\n");
}
return 0;
}