【id:56】【20分】C. 子串循环问题 (Ver. I)

题目描述

给定一个字符串，求需要添加至少几个字符到字符串末尾才能使得整个字符串串由某一个不为本身的子串循环构成？
如"abca"，添加"bc"后构成"abcabc"，其由子串"abc"循环构成;也可以添加"abca"后构成"abcaabca"，其由子串"abca"循环构成，相比之下"bc"只有2个字符，添加的字符量最少。

输入

第一行包括一个整数T(1 <= T <= 100)，代表测试组数

每组测试数据包括一行字符串，其长度范围为 [3, 104]

输出

对于每组测试数据

输出一个整数N，代表添加的最小字符数量

输入样例1

5
aaa
abca
abcdefg
abcabcabca
qweaqwec

输出样例1

0
2
7
2
8

以下为思路

先是字串循环的定律。

设该字符串为str，长度为len。经过kmp算法处理后可得到一个next[ ]数组，数组中共有len个元素（这里初值next[0]为-1）。str存在最小循环节，长度L为len-next[len];

如果len可以被len-next[len]整除(即len%(len-next[len])==0)，说明该字符串由循环字节组成，如aaa，abcdefg。若next[len]=0，说明只需要再尾部加上一个该字符串的最小循环节即可。abcdefg尾部加上它自己即可，此时返回len；如果不为0说明只能加上它自己才能构成一个字串循环（比如aaa只能加上a，abc同理），此时返回0即可。

如果不能被整除，说明需要补上几个字母才行（比如abca需要补上bc才能构成循环），

需要补的字母个数为len-next[len]-(next[len]%(len-next[len]))个。

以下为代码

这题有很多方法，直接暴力也可以解。这里用的kmp。

#include <iostream>
using namespace std;

void kmp(string str)
{
	int k = -1;
	int j = 0;
	int len = str.length();
	int* next = new int[len+1];
	for (int i = 0;i <= len;i++)
	{
		next[i] = 0;
	}
	next[0] = -1;
	while (j < len)
	{
		if (k == -1 || str[k] == str[j])
		{
			next[++j] = ++k;
		}
		else
		{
			k = next[k];
		}
	}
	int t = next[len];
	if (len % (len - t) == 0)
	{
		if (next[len] == 0)
		{
			cout << len;
		}
		else
		{
			cout << 0;
		}
	}
	else
	{
		cout << len-t-(t%(len-t));
	}
	cout << endl;

}
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		string str;
		cin >> str;
		kmp(str);
	}
	return 0;
}