Codeforces 827C DNA Evolution(多维树状数组)

题意：

给定一个只包含A,T,C,G的字符串,有如下两种操作

1）修改一个点的字母

2）给定区间L, R和一个字符串e (strlen(e) <=10)，组成一个足够区间长度的由若干个e重复组成的新串，eee...，问L，R区间中有几个位置对应的字母跟这个新的字符串对应的相同。

思路：

因为e的长度最多才10，所以对于一个字母，它最多有10个不同的起始位置，10个不同的长度进行重复，所以建立一个4*10*10的树状数组，再维护就可以了。

注意维护和查询时候的细节就好了，并不难。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
char s[maxn], e[11];
map<char, int> mp;
int c[11][11][4][maxn], cnt[11][11];
int n, len;
int lowbit(int x) {return x & -x;}
void update(int x, int y, int z, int pos, int val)
{
while(pos <= cnt[x][y])
{
c[x][y][z][pos] += val;
pos += lowbit(pos);
}
}
int query(int x, int y, int z, int pos)
{
int ans = 0;
while(pos)
{
ans += c[x][y][z][pos];
pos -= lowbit(pos);
}
return ans;
}
void init()
{
memset(c, 0, sizeof c);
mp['A'] = 0, mp['T'] = 1, mp['G'] = 2, mp['C'] = 3;
len = strlen(s+1);
for(int i = 1; i <= min(10, len); ++i)
for(int j = i; j <= 10; ++j)
{
cnt[i][j] = (len-i)/j+1;
for(int k = 0; i+k*j <= len; ++k)
{
if(s[i+k*j] == 'A') update(i, j, 0, k+1, 1);
if(s[i+k*j] == 'T') update(i, j, 1, k+1, 1);
if(s[i+k*j] == 'G') update(i, j, 2, k+1, 1);
if(s[i+k*j] == 'C') update(i, j, 3, k+1, 1);
}
}
}
void change(int t, char e)
{
int mmp = min(min(10, len), t);
for(int i = 1; i <= mmp; ++i)
for(int j = i; j <= 10; ++j)
{
if((t-i)%j == 0)
{
int k = (t-i)/
4000
j;
update(i, j, mp[s[t]], k+1, -1);
update(i, j, mp[e], k+1, 1);
}
}
s[t] = e;
}
int main()
{
ios::sync_with_stdio(0);
cin >> s+1;
init();
cin >> n;
int key, x, l, r;
for(int i = 1; i <= n; ++i)
{
cin >> key;
if(key == 1)
{
cin >> x;
cin >> e;
change(x, e[0]);
}
if(key == 2)
{
cin >> l >> r;
cin >> e+1;
int ans = 0, elen = strlen(e+1);
for(int j = 1; l+j-1 <= r && j <= elen; ++j)
{
int start = (l+j-1)%elen;
if(start == 0) start = elen;
if(l-1-start >= 0)
{
int ll = (l-1-start)/elen+1;
ans -= query(start, elen, mp[e[j]], ll);
}
if(r-start >= 0)
{
int rr = (r-start)/elen+1;
ans += query(start, elen, mp[e[j]], rr);
}
}
cout << ans << endl;
}
}
return 0;
}

继续加油~