DNA Evolution CodeForces - 828E（树状数组）

题中有两种操作，第一种把某个位置的字母修改，第二种操作查询与[L, R]内与给出字符串循环起来以后对应位置的字母相同的个数。给出的字符串最大长度是10。

用一个四维树状数组表示

cnt[ATCG的编号][可能的给出字符串的长度][原字符中这个位置在循环中的位置][从开始到这个位置] = 出现的次数。

然后对于一开始的字符串，暴力更新每一种情况。

对于1操作，对于每一种可能的循环节长度，把这个位置的贡献清除，然后换成要填入的字母

对于2操作，（l+i）表示在原字符串的位置，(l+i-1)%len表示这个位置在给出的字符串中循环的位置，然后对于给出字符串的每一个 i 位置，查询对应的个数有多少个，相加。

#include<map>
#include<set>
#include<ctime>
#include<cmath>
#include<stack>
#include<queue>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define lowbit(x) (x & (-x))
#define INOPEM freopen("in.txt", "r", stdin)
#define OUTOPEN freopen("out.txt", "w", stdout)

typedef unsigned long long int ull;
typedef long long int ll;
const double pi = 4.0*atan(1.0);
const int inf = 0x3f3f3f3f;
const int maxn = 1e5+;
const int maxm = ;
const int mod = 1e9+;
using namespace std;

int n, m;
int T, tol;
int cnt[][][][maxn];
char s[maxn];
int mx;

int getid(char c) {
    if(c == 'A')    return ;
    if(c == 'T')    return ;
    if(c == 'C')    return ;
    if(c == 'G')    return ;
}

void update(int id, int i, int j, int pos, int val) {
    for(int x=pos; x<=mx; x+=lowbit(x))    cnt[id][i][j][x] += val;
}

int query(int id, int i, int j, int pos) {
    int ans = ;
    for(int x=pos; x>; x-=lowbit(x))    ans += cnt[id][i][j][x];
    return ans;
}

void init() {
    for(int i=; i<=; i++) {
        for(int j=; j<=mx; j++) {
            update(getid(s[j]), i, (j-)%i, j, );
        }
    }
}

int main() {
    scanf("%s", s+);
    mx = strlen(s+);
    memset(cnt, , sizeof cnt);
    init();
    int q;
    scanf("%d", &q);
    while(q--) {
        int op;
        scanf("%d", &op);
        if(op == ) {
            int pos;
            char tmp[];
            scanf("%d%s", &pos, tmp);
            for(int i=; i<=; i++)    update(getid(s[pos]), i, (pos-)%i, pos, -);
            for(int i=; i<=; i++)    update(getid(tmp[]), i, (pos-)%i, pos, );
            s[pos] = tmp[];
        } else {
            int l, r;
            char tmp[];
            scanf("%d%d%s", &l, &r, tmp);
            int len = strlen(tmp);
            int ans = ;
            for(int i=; i<len; i++) {
                ans += query(getid(tmp[i]), len, (l+i-)%len, r) - query(getid(tmp[i]), len, (l+i-)%len, l-);
            }
            printf("%d\n", ans);
        }
    }
    return ;
}