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)/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;}

继续加油~