网上有篇题解写的是线段树合并维护求值？
题目描述
有一个只包含小写字母，长度为 $n$ 的字符串 $S$ 。有一些字母是好的，剩下的是坏的。
定义一个子串 $S_{lldots r}$是好的，当且仅当这个子串包含不超过 $k$ 个坏的字母。
求有多少个不同的满足以下要求的字符串 $T$ ：
- $T$ 作为 $S$ 的子串出现过。
- 存在一个 $T$ 出现的位置 $[l,r]$ ，满足 $S_{lldots r}$ 是好的。
输入格式
第一行有一个字符串 $S$ 。
第二行有一个字符串 $B$ 。若 $B_i=‘1’$ 则表示 $S_i$ 是好的，否则表示 $S_i$ 是坏的。
第三行有一个整数 $k$ 。
输出格式
一个整数：答案。
数据范围与提示
子任务 $1$（$10$ 分）：$nleq 10$。
子任务 $2$（$10$ 分）：$nleq 100$。
子任务 $3$（$10$ 分）：$nleq 1000$。
子任务 $4$（$10$ 分）：$nleq 100000,k=n$。
子任务 $5$（$10$ 分）：$nleq 100000,k=0$。
子任务 $6$（$20$ 分）：$nleq 100000$，若$S_i=S_j$，则$B_i=B_j$。
子任务 $7$（$30$ 分）：$nleq 100000$。
对于 $100\%$ 的数据：$1leq nleq {10}^5,0leq kleq {10}^5$， $S$ 只包含小写字母。
题目来源：全是水题的GDOI模拟赛 by yww
题目分析
定位：比较模板的SAM题（然而我一个月前并不会SAM）
题意即求满足一定条件的若干个字符串里本质不同的子串个数。当然这里的“一定条件”比较特殊，是连续的一段子串。（如果这里的“一定条件”字符串是若干个互不相干的串，似乎就需要“广义后缀自动机”来处理了）
既然对于每一个新增的节点，其合法的子串都有一个左边界，那么在SAM里处理的时候，就可以对每一节点加一个权值$mx[u]$表示该节点的最长合法扩展长度。这个权值的作用就在于建完自动机后的$calc()$，原先数本质不同的子串个数是这样的： ans += len[p]-len[fa[p]]; 现在就是 ans += std::max(std::min(mx[p], len[p])-len[fa[p]], 0); 。记得注意一下extend里对mx的转移。
1 #include<bits/stdc++.h> 2 const int maxn = 200035; 3 4 int n,lim,sum[maxn],size[maxn],cnt[maxn],pos[maxn]; 5 long long ans; 6 struct SAM 7 { 8 int ch[maxn][27],fa[maxn],len[maxn],mx[maxn],lst,tot; 9 void init() 10 { 11 lst = tot = 1; 12 } 13 void extend(int c, int v) 14 { 15 int p = lst, np = ++tot; 16 lst = np, len[np] = len[p]+1, mx[np] = v; //len[p+1] Here 居然打成标红的这个 17 for (; p&&!ch[p][c]; p=fa[p]) ch[p][c] = np; 18 if (!p) fa[np] = 1; 19 else{ 20 int q = ch[p][c]; 21 if (len[p]+1==len[q]) fa[np] = q; 22 else{ 23 int nq = ++tot; 24 len[nq] = len[p]+1, mx[nq] = mx[q]; 25 memcpy(ch[nq], ch[q], sizeof ch[q]); 26 fa[nq] = fa[q], fa[q] = fa[np] = nq; 27 for (; p&&ch[p][c]==q; p=fa[p]) 28 ch[p][c] = nq; 29 } 30 } 31 } 32 void calc() 33 { 34 for (int i=1; i<=tot; i++) ++cnt[len[i]]; 35 for (int i=1; i<=tot; i++) cnt[i] += cnt[i-1]; 36 for (int i=1; i<=tot; i++) pos[cnt[len[i]]] = i, --cnt[len[i]]; 37 for (int i=tot; i; i--) 38 { 39 int p = pos[i]; 40 mx[fa[p]] = std::max(mx[fa[p]], mx[p]); 41 ans += std::max(std::min(mx[p], len[p])-len[fa[p]], 0); 42 } 43 } 44 }f; 45 char s[maxn],t[maxn]; 46 47 int main() 48 { 49 scanf("%s%s%d",s+1,t+1,&lim); 50 n = strlen(s+1); 51 f.init(); 52 for (int i=1, j=0; i<=n; i++) 53 { 54 sum[i] = (t[i]=='0')+sum[i-1]; 55 while (sum[i]-sum[j] > lim) ++j; 56 f.extend(s[i]-'a'+1, i-j); 57 } 58 f.calc(); 59 printf("%lld ",ans); 60 return 0; 61 }
END