计蒜客-2017 ACM-ICPC 亚洲区(乌鲁木齐赛区)网络赛G题Query on a strin(树状数组+暴力更新)

题意：

给一个<=1e5的主串，一个<=10的匹配串，n次询问或操作，询问主串的[l, r]区间内与匹配串能匹配的个数(可重叠)，操作给定一个位置pos和字符ch，将s[pos]改为ch。

思路：

由于匹配串长度很小，完全可以暴力去做，主串中与匹配串匹配的子串只有起点位置有值，然后通过树状数组维护区间和即可，不过查询时要注意一个小判断，区间有可能不足匹配串的长度。更新时，暴力去更新，改变一个字符，只会影响其前面少于十个位置，其后面少于十个位置。

代码：

#include <bits/stdc++.h>  
using namespace std;  
const int maxn = 1e5+5;  
char s[maxn], p[15], mk1, mk2;  
int a[maxn], tree[maxn];  
int t, n, len1, len2, x, y;  
int lowbit(int x){return x&(-x);}  
void update(int k, int val)  
{  
    while(k <= len1)  
    {  
        tree[k] += val;  
        k += lowbit(k);  
    }  
}  
int query(int k)  
{  
    int ans = 0;  
    while(k)  
    {  
        ans += tree[k];  
        k -= lowbit(k);  
    }  
    return ans;  
}  
void init()  
{  
    len1 = strlen(s+1);  
    len2 = strlen(p+1);  
    memset(tree, 0, sizeof tree);  
    for(int i = 1; i+len2-1 <= len1; ++i)  
    {  
        int t = 1;  
        while(t <= len2 && s[i+t-1] == p[t]) ++t;  
        a[i] = (t > len2);  
        update(i, a[i]);  
    }  
}  
int main()  
{  
    //freopen("in.txt", "r", stdin);  
    for(scanf("%d", &t); t--;)  
    {  
        scanf("%d", &n);  
        scanf("%s", s+1);  
        scanf("%s", p+1);  
        init();  
        while(n--)  
        {  
            scanf(" %c %d", &mk1, &x);  
            if(mk1 == 'C')  
            {  
                scanf(" %c", &mk2);  
                if(s[x] == mk2) continue;  
                s[x] = mk2;  
                int st = x-len2+1, ed = x+len2-1;  
                if(st < 1) st = 1;  
                if(ed > len1) ed = len1;  
                for(int i = st; i <= ed && i+len2-1 <= len1; ++i)  
                {  
                    int t = 1;  
                    while(t <= len2 && s[i+t-1] == p[t]) ++t;  
                    int chg = (t > len2);  
                    if(a[i]) a[i] = 0, update(i, -1);  
                    if(chg) a[i] = 1, update(i, 1);  
                }  
            }  
            else  
            {  
                scanf("%d", &y);  
                int ans = 0;  
                if(y-len2+1 >= x)   
                    printf("%d\n", query(y-len2+1)-query(x-1));  
                else puts("0");  
            }  
        }  
        puts("");
    }  
    return 0;  
}

继续加油~