Codeforces 827C - DNA Evolution

827C - DNA Evolution

思路：

写4*10*10个树状数组，一个维度是4（ATCG），另一个维度是长度len，另一个维度是pos%len，因为两个pos,如果len和pos%len相同，那么它们就在一个树状数组里。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e5+;
int a[][][][N];
int n;
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int pos,int c,int d)
{
    while(x<=n)
    {
        for(int i=;i<=;i++)
        a[c][i][pos%i][x]+=d;
        x+=lowbit(x);
    }
}
int sum(int x,int pos,int c,int len)
{
    int ans=;
    while(x>)
    {
        ans+=a[c][len][pos%len][x];
        x-=lowbit(x);
    }
    return ans;
}
int mp[];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie();
    string s,t;
    int m,a,l,r;
    mp['A']=;
    mp['T']=;
    mp['C']=;
    mp['G']=;
    cin>>s;
    n=s.size();
    for(int i=;i<s.size();i++)
    {
        update(i+,i+,mp[s[i]],);
    }
    cin>>m;
    while(m--)
    {
        cin>>a;
        //cout<<1<<endl;
        if(a==)
        {
            cin>>l>>r>>t;
            int ans=;
            for(int i=;i<t.size();i++)
            ans+=sum(r,i+l,mp[t[i]],t.size())-sum(l-,i+l,mp[t[i]],t.size());
            cout<<ans<<endl;
        }
        else
        {
            cin>>l;
            cin>>t;
            update(l,l,mp[s[l-]],-);
            update(l,l,mp[t[]],);
            s[l-]=t[];
        }
    }
    return ;
}