2017 ACM-ICPC乌鲁木齐网络赛 G. Query on a string（KMP+树状数组）

题目链接：https://www.jisuanke.com/contest/870

题意：

给出两个字符串S和T，Q次操作：①C a b表示将第a个字符改为b；②Q l r表示T在S的子串[l, r]中共出现多少次

其中|S|<=100000，|T|<=10，Q<=100000

思路：

若S串的子串[x-|T|+1, x]和T串完美匹配，那么flag[x]==1，否则flag[x]==0，sum[]是flag[]的前缀和

这样每次查询区间[l, r]答案就是sum[r]-sum[l+|T|-2]，注意特判r<l+|T|-2的情况

用树状数组维护就好

又因为|T|<10，所以每次修改影响的范围很小，可以直接暴力

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define LL long long
char str[100005], jud[100005];
int n, nj[100005], ns[100005];
LL tre[100005];
void Update(int x, int val)
{
while(x<=n)
{
tre[x] += val;
x += x&-x;
}
}
LL Query(int x)
{
LL sum = 0;
while(x)
{
sum += tre[x];
x -= x&-x;
}
return sum;
}
int main(void)
{
char ch[15];
int T, Q, m, p, q, L, R;
scanf("%d", &T);
while(T--)
{
scanf("%d%s%s", &Q, str+1, jud+1);
n = strlen(str+1);
m = strlen(jud+1);
p = 0, q = 1;
memset(nj, 0, sizeof(nj));
memset(tre, 0, sizeof(tre));
memset(ns, 0, sizeof(ns));
while(q<=m)
{
if(p==0 || jud[p]==jud[q])
{
p++, q++;
if(jud[p]==jud[q])
nj[q] = nj[p];
else
nj[q] = p;
}
else
p = nj[p];
}
p = q = 1;
while(p<=n)
{
while(str[p]!=jud[q] && q>=1)
q = nj[q];
if(q==0)
{
ns[p] = 0;
p++, q++;
continue;
}
ns[p++] = q++;
if(q==m+1)
{
Update(p-1, 1);
q = nj[q];
}
}
while(Q--)
{
scanf("%s", ch);
if(ch[0]=='Q')
{
scanf("%d%d", &L, &R);
if(L+m-1>R)
printf("0\n");
else
printf("%lld\n", Query(R)-Query(L+m-2));
}
else
{
scanf("%d%s", &R, ch);
p = max(R-15, 0), q = ns[p];
str[R] = ch[0];
while(p<=R+15 && p<=n)
{
while(str[p]!=jud[q] && q>=1)
q = nj[q];
if(q==0)
{
if(ns[p]==m)
Update(p, -1);
ns[p] = 0;
p++, q++;
continue;
}
if(ns[p]==m)
Update(p, -1);
ns[p++] = q++;
if(q==m+1)
{
Update(p-1, 1);
q = nj[q];
}
}
}
}
printf("\n");
}
return 0;
}