多个树状数组 DNA Evolution CodeForces - 828E

题目

题意：给你一个DNA串，有两种操作

1 a e：把位置a的字符换成字符e

2 a b e：问[ a, b ]的区间内eeeeee..（e循环串）与相同的的字符有几个，并输出。

看题解，然后自己写，这题做了一下午，原因还是因为自己对于每个位置表示的东西不够熟悉，老是东错西错。。

思路：用线段树来保存每个字母在每个区间处于每种循环串位置的数量，
node[ 字母种类 ] [ e的长度 ] [ 长度取模(相对e的长度位置) ] [ 位置 ],然后就是基本的树状数组插入和查询，注意细节。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn = 1e5+10;
int node[4][11][11][maxn];  
int h[110];
char s[maxn], e[11];

int lowbit(int x){
    return x&(-x);
}

void add(int ty, int pos, int x, int d){  //ty：字母的种类 pos:相对位置 x:实际位置 d：变换的值
    for(; x < maxn; x+=lowbit(x)){
        for(int k = 1; k <= 10; k++){
            node[ty][k][pos%k][x] += d;
        }
    }
}

int sum(int ty, int len,int pos, int x){  //ty:字母的种类 len:e子串的长度 pos：相对于e子串的位置 x:实际位置
    int ans = 0;
    for(; x > 0 ; x-=lowbit(x)){
        ans += node[ty][len][pos][x];
    }
    return ans;
}

int query(int l,int r,int ty,int len,int pos){  //区间[l,r]
    return sum(ty, len, pos, r) - sum(ty, len, pos, l-1);
}

int main(){
    int n;
    h['A'] = 0; h['T'] = 1; h['C'] = 2; h['G'] = 3;  //给四个字母标记数字
    scanf("%s%d",s,&n);
    int len = strlen(s);
    for(int i = 0;i < len; i++){
        add(h[s[i]], i, i+1, 1);  //因为树状数组是从1开始，则i+1为实际位置。但是相对位置之后要取模所以不能+1，就按照原来的位置取模完才能会是正确的相对位置
    }
    while(n--){
        int a, b, c;
        scanf("%d",&a);
        if(a == 1){
            scanf("%d%s",&b,e);
            add(h[s[b-1]], b-1, b, -1);  //把s[b-1]原来的字母的数量减去
            add(h[e[0]], b-1, b, 1);   //把新的字母的数量+1
            s[b-1] = e[0];
        }
        else{
            scanf("%d%d%s",&b,&c,e);
            len = strlen(e);
            int ans = 0;
            for(int i = 0; i < len; i++){  //枚举每个e字符串的上的字符的数量。
                ans += query(b, c, h[e[i]], len, (i + b -1)%len);  //这里几个（i+b-1）要记得取模len，不然就凉了
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}