給定字符串 S(父親名+母親名拼接),求所有**既是前綴又是後綴**的子串長度,按遞增輸出。
多測試用例,|S| ≤ 400,000,僅小寫字母。
next 數組(前綴函數)next[i] = 模式串 [0..i-1] 的最長真前綴同時為後綴的長度。S 計算 next,則 next[|S|] 為最長邊界長度。len = next[len] 可得到次長邊界,直到 0。|S|(整串自身)。#include
using namespace std;
const int MAXN = 400005;
char s[MAXN];
int nxt[MAXN], ans[MAXN];
void getNext(const char* s, int len) {
int i = 0, j = -1; nxt[0] = -1;
while (i < len) {
if (j == -1 || s[i] == s[j]) nxt[++i] = ++j;
else j = nxt[j];
}
}
int main() {
while (scanf("%s", s) == 1) {
int len = strlen(s);
getNext(s, len);
int cnt = 0, tmp = nxt[len];
while (tmp > 0) ans[++cnt] = tmp, tmp = nxt[tmp];
for (int i = cnt; i >= 1; --i) printf("%d ", ans[i]);
printf("%d\n", len);
}
return 0;
}
O(|S|) 每測試用例。O(|S|)。next 數組定義與標準 KMP 一致,下標從 0 開始,nxt[0] = -1 便於統一邏輯。len 始終是邊界,最後補上。