[TJOI2019]甲苯先生和大中锋的字符串

Description#

给定模式串 s 和一个整数 k ，求所有出现次数为 k 次的字符串中长度出现次数最多的长度。

多测。

$n\le {10}^5,\sum n\le 3\cdot {10}^6$

Solution#

这些一看起来暴力就很劣的字符串可能多半和 SAM 有点关系。

而且这题个人感觉比较萌萌，主要是他让我们干的事全部写的明明白白。

所以说我们要先能知道那些字符串出现次数为 k ，维护 SAM 上状态的 siz 就可以了，那么对于每个状态如果满足条件，所有包含在这个长度里的字符串都有 1 的贡献，区间加，差分，最后统一统计。

Code#

Code

Copy

/*

*/
#include 
using namespace std;
typedef long long ll;
const int N = 6e6 + 10;
int k, arr[N];
inline int read() {
	char ch = getchar();
	int s = 0, w = 1;
	while (!isdigit(ch)) {if (ch == '-') w = -1; ch = getchar();}
	while (isdigit(ch)) {s = (s << 3) + (s << 1) + (ch ^ 48); ch = getchar();}
	return s * w;
}
struct SAM {
	int n, cnt, las, len[N], link[N], ch[N][26];
	char s[N]; int tong[N], rk[N], siz[N];
	inline void init() {cnt = las = 1; memset(ch[1], 0, sizeof(ch[1]));}
	inline void SAM_stru(int c) {
		int cur = ++cnt, p = las;
		memset(ch[cur], 0, sizeof(ch[cur]));
		las = cur;
		len[cur] = len[p] + 1; siz[cur] = 1;
		while (p && !ch[p][c]) ch[p][c] = cur, p = link[p];
		if (!p) {link[cur] = 1; return ;}
		int q = ch[p][c];
		if (len[p] + 1 == len[q]) {link[cur] = q; return ;}
		int clo = ++cnt;
		link[clo] = link[q]; len[clo] = len[p] + 1;
		link[q] = link[cur] = clo; siz[clo] = 0;
		memcpy(ch[clo], ch[q], sizeof(ch[clo]));
		while (p && ch[p][c] == q) ch[p][c] = clo, p = link[p];
	}
	inline void Tong_sort() {
		for (int i = 1; i <= cnt; ++i) tong[i] = 0;
		for (int i = 1; i <= cnt; ++i) ++tong[len[i]];
		for (int i = 1; i <= cnt; ++i) tong[i] += tong[i - 1];
		for (int i = 1; i <= cnt; ++i) rk[tong[len[i]]--] = i;
		for (int i = cnt, u, v; i; --i) {
			v = rk[i]; u = link[v];
			siz[u] += siz[v];
			if (siz[v] == k) ++arr[len[u] + 1], --arr[len[v] + 1];
		}
	}
} s;
inline void mian() {
	s.init();
	scanf("%s", s.s + 1); s.n = strlen(s.s + 1); k = read();
	for (int i = 1; i <= s.n; ++i) s.SAM_stru(s.s[i] - 'a');
	s.Tong_sort();
	int mx = -1, ans = -1;
	for (int i = 1; i <= s.n; ++i) {
		arr[i] += arr[i - 1];
		if (mx <= arr[i]) {
			mx = arr[i]; ans = i;
		}
	}
	printf("%d\n", (mx > 0) ? ans : -1);
	for (int i = 1; i <= s.n + 1; ++i) arr[i] = 0;
}
int main() {
	int t = read();
	while (t--) mian();
	return 0;
}