题目大意:
就是现在初始的时候给出一个长度为n的串S(n <= 50000), 给定整数K, 然后有m次操作或询问 (m <= 200000), 对于每一次操作或询问, 用1表示在当前的S后面加上一个字母形成新的S, 操作2就是询问当前的字符串中有多少个K串, 当操作是1的时候数字后面会有1个字母表示要添加的字符
K串指的是S的子串, 且在S中出现了不少于K次
大致思路:
首先不难想到后缀自动机, 对于初始的串S建立后缀自动机, 然后每个状态的Right集合的大小是否大于等于K就代表了这个状态代表的子串是否是K串
然后由于后缀自动机的在线性质, 每次添加新的字符到S后面时, 将新的字符插入后缀自动机, 然后考虑新增加的np状态对于Parent书中其父亲结点的Right集合大小的影响即可
写的时候注意一些细节, 刚开始没有注意到nq和q对应状态见得关系WA了几次
代码如下:
Result : Accepted Memory : 26064 KB Time : 1684 ms
/* * Author: Gatevin * Created Time: 2015/4/13 14:28:04 * File Name: Rin_Tohsaka.cpp */ #include<iostream> #include<sstream> #include<fstream> #include<vector> #include<list> #include<deque> #include<queue> #include<stack> #include<map> #include<set> #include<bitset> #include<algorithm> #include<cstdio> #include<cstdlib> #include<cstring> #include<cctype> #include<cmath> #include<ctime> #include<iomanip> using namespace std; const double eps(1e-8); typedef long long lint; #define foreach(e, x) for(__typeof(x.begin()) e = x.begin(); e != x.end(); ++e) #define SHOW_MEMORY(x) cout<<sizeof(x)/(1024*1024.)<<"MB"<<endl #define maxm 250100 #define maxn 500100 #define maxl 50010 int n, K, m; char s[maxl]; lint ans; struct Suffix_Automation { struct State { State *par; State *go[26]; int val, mi, cnt, right; void init(int _val = 0) { val = _val, par = 0, mi = cnt = right = 0; memset(go, 0, sizeof(go)); } int calc() { if(par == 0) return 0; else return val - par->val; } }; State *root, *last, *cur; State nodePool[maxn]; State* newState(int val = 0) { cur->init(val); return cur++; } void initSAM() { cur = nodePool; root = newState(); last = root; } void extend(int w, bool flag) { State *p = last; State *np = newState(p->val + 1); np->right = 1; while(p && p->go[w] == 0) { p->go[w] = np; p = p->par; } if(p == 0) { np->par = root; } else { State *q = p->go[w]; if(q->val == p->val + 1) { np->par = q; } else { State *nq = newState(p->val + 1); memcpy(nq->go, q->go, sizeof(q->go)); nq->par = q->par; if(flag) nq->right = q->right; //如果是后来的逐一添加的话就同时更新nq的right //不是的话因为会执行topo和SAMInfo函数, 不要更新nq的right q->par = nq; np->par = nq; while(p && p->go[w] == q) { p->go[w] = nq; p = p->par; } } } last = np; } int d[maxm]; State *b[maxn]; void topo() { int cnt = cur - nodePool; int maxVal = 0; memset(d, 0, sizeof(d)); for(int i = 1; i < cnt; i++) maxVal = max(maxVal, nodePool[i].val), d[nodePool[i].val]++; for(int i = 1; i <= maxVal; i++) d[i] += d[i - 1]; for(int i = 1; i < cnt; i++) b[d[nodePool[i].val]--] = &nodePool[i]; b[0] = root; } void SAMInfo() { int cnt = cur - nodePool; for(int i = cnt - 1; i > 0; i--) { b[i]->par->right += b[i]->right; b[i]->mi = b[i]->par->val + 1; } } void update()//每次新加入结尾的np状态会影响到Parent树上其所有的父亲 { State *np = last; while(np != root) { if(np->right >= K) ans += np->val - np->par->val; if(np->par->right >= K) break;//再往上的点没有必要, 因为已经记过数了 np->par->right++; np = np->par; } } }; Suffix_Automation sam; int main() { while(scanf("%d %d %d",&n, &m, &K) != EOF) { getchar(); gets(s); sam.initSAM(); for(int i = 0; i < n; i++) sam.extend(s[i] - 'a', 0); sam.topo(); sam.SAMInfo(); ans = 0; int cnt = sam.cur - sam.nodePool; for(int i = 1; i < cnt; i++) if(sam.b[i]->right >= K) ans += sam.b[i]->val - sam.b[i]->mi + 1; int op; char ad; while(m--) { scanf("%d", &op); if(op == 2) printf("%I64d\n", ans); else { scanf(" %c", &ad); sam.extend(ad - 'a', 1); sam.update(); } } } return 0; }