Query on a string【线段树】

题意：两个字符串，1.问在第一个字符串的a~b中出现几次第二个字符串，2.修改第一个字符串的某个位置

思路：用s数组记录从当前位置开始能不能匹配，能匹配用1表示，用线段树记录，查询a~b，只要询问a ~ b-len2+1 中有几个1，注意边界。修改的话，最多影响s数组中的10个位置，暴力修改下即可。

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<list>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<algorithm>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e5+5;
#define lson(x) 2*x
#define rson(x) 2*x+1

struct data
{
	int l,r,num;
}node[maxn*4];
char p[maxn],T[maxn];
int len1,len2,f[maxn],s[maxn];

void get(int l,int r)
{
    for(int i=l,j=0;i<=r;i++)
    {
        while(j&&p[j]!=T[i]) j=f[j];
        if(p[j]==T[i]) j++;
        if(j==len2)
		{
			s[i-len2+1] = 1;
			j = f[j]; 
		} 
    }
}

void getf()
{
    f[0]=0,f[1]=0;
    for(int i=1;i<len2;i++)
    {
        int j=f[i];
        while(j&&p[i]!=p[j])j=f[j];
        f[i+1]=(p[i]==p[j]?j+1:0);
    }
}

void pushup(int cnt)
{
	node[cnt].num = node[lson(cnt)].num + node[rson(cnt)].num;
}

void build(int x,int y, int cnt)
{
	node[cnt].l = x;
	node[cnt].r = y;
	if(x == y)
	{
		node[cnt].num = s[x-1];
		return;
	}
	int mid = (x+y) / 2;
	build(x,mid,lson(cnt));
	build(mid+1,y,rson(cnt));
	pushup(cnt);
}

void up(int x,int y,int cnt,int val)
{
	if(x == node[cnt].l && y == node[cnt].r)
	{
		node[cnt].num = val;
		return;
	}
	int mid = (node[cnt].l + node[cnt].r) / 2;
	if(y <= mid)
		up(x,y,lson(cnt),val);
	else
		up(x,y,rson(cnt),val);
	pushup(cnt);
}

int fid(int x,int y,int cnt)
{
	if(x == node[cnt].l && y == node[cnt].r)
		return node[cnt].num;
	int mid = (node[cnt].l + node[cnt].r) / 2;
	if(y <= mid)
		return fid(x,y,lson(cnt));
	else if(x >= mid+1)
		return fid(x,y,rson(cnt));
	else
		return fid(mid+1,y,rson(cnt)) + fid(x,mid,lson(cnt));
}

int main(void)
{
	int t,n,a,b;
	char op[5],ch[5];
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		scanf("%s%s",T,p);
		len1 = strlen(T);
		len2 = strlen(p);
		memset(s,0,sizeof s);
		getf();
		get(0,len1-1);
		build(1,len1,1);
		for(int i = 0; i < n; i++)
		{
			scanf("%s",op);
			if(op[0] == 'C')
			{
				scanf("%d%s",&a,ch);
				T[a-1] = ch[0];
				int st = max(1,a-len2+1);
				for(int k = st; k <= a && k+len2-1 <= len1; k++)
				{
					int flag = 1;
					for(int j = k-1,jj = 0; jj < len2; j++,jj++)
					{
						if(T[j] != p[jj])
						{
							flag = 0;
							break;
						}
					}
					if(flag != s[k-1])
					{
						up(k,k,1,flag);
						s[k-1] ^= 1; 
					}	
				}
			}
			else
			{
				scanf("%d%d",&a,&b);
				if(b-a+1 < len2)
					printf("0\n");
				else
					printf("%d\n",fid(a,b-len2+1,1));
			}
		}
		putchar('\n');
	}
	return 0;
}
/*
1
5
AABBABA
AA
Q 1 3
C 6 A
Q 2 7
C 2 B
Q 1 5
*/