1. 程式人生 > >51nod 1286 三段子串(樹狀數組+拓展kmp)

51nod 1286 三段子串(樹狀數組+拓展kmp)

for names 每次 color ack har -a || clas

題意:

給定一個字符串S,找到另外一個字符串T,T既是S的前綴,也是S的後綴,並且在中間某個地方也出現一次,並且這三次出現不重合。求T最長的長度。

例如:S = "abababababa",其中"aba"既是S的前綴,也是S的後綴,中間還出現了一次,並且同前綴後綴均不重合。所以輸出"aba"的長度3。如果找不到一個符合條件的字符,輸出0。 題解: 先對S做一次拓展kmp,求出next數組 然後其實next每一個數就對應了一個前綴匹配的情況。 比如說next[3]和next[5]都為2,就意味著前綴為2的子串可以和位置為3和位置為5的子串匹配。 然後我們按照next數組倒序來做。 對於長度為i的前綴,先判斷是否存在長度為i的後綴匹配,如果存在再詢問是否存在中間子串。 每次匹配的結果都插入到樹狀數組裏面(因為如果某個位置可以匹配L的前綴,那麽實際上1~L-1都能匹配) 然後詢問i+1 ~ L-2i+1這個區間裏有沒有能匹配的子串(即中間的那個串)。
#include <iostream>
#include 
<cstring> #include <vector> using namespace std; const int maxn = 1e6 + 100; char S[maxn]; int Next[maxn], extend[maxn]; vector<int> G[maxn]; int len; int c[maxn*4]; void Modify(int x, int s){ for(; x <= len; x += x&(-x)) c[x] += s; } int Query(int y){ if(y <= 0) return
0; int ans = 0; for(int x = y; x; x -= x&(-x)) ans += c[x]; return ans; } int query(int x, int y) { return x <= y ? Query(y) - Query(x-1) : 0; } void GetNext(char *T, int *Next){ int len = strlen(T); Next[0] = len; int a, p; for(int i = 1, j = -1; i < len; i++, j--){
if(j < 0 || i + Next[i-a] >= p){ if(j < 0) p = i, j = 0; while(p < len && T[p] == T[j]) p++, j++; Next[i] = j; a = i; } else Next[i] = Next[i-a]; } } void GetExtend(char *S, char *T, int *extend, int *Next){ GetNext(T, Next); int a, p; int slen = strlen(S), tlen = strlen(T); for(int i = 0, j = -1; i < slen; i++, j--){ if(j < 0 || i + Next[i-a] >= p){ if(j < 0) p = i, j = 0; while(p < slen && j < tlen && S[p] == T[j]) p++, j++; extend[i] = j; a = i; } else extend[i] = Next[i-a]; } } int main() { cin>>S; len = strlen(S); GetNext(S, Next); //for(int i = 0; i < len; i++) cout<<Next[i]<<" "; cout<<endl; for(int i = 0; i < len; i++) G[Next[i]].push_back(i); int ans = 0; for(int i = len-1; i >= 1; i--){ for(auto x : G[i]) Modify(x+1, 1); int n = G[i].size(); if(n == 0) continue; if(G[i][n-1] != len-i) continue; if(query(i+1, len-2*i+1)) { ans = i; break; } } cout<<ans<<endl; return 0; }

51nod 1286 三段子串(樹狀數組+拓展kmp)