Codeforces Round #423 (Div. 1, rated, based on VK Cup Finals)

Codeforces Round #423 (Div. 1, rated, based on VK Cup Finals)

A.String Reconstruction

B. High Load

C. DNA Evolution

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

1）修改一个点的字母.

2）给定一个字符串e ($left | e ight |leq 10$)，生成一个由e重复组成的新串，eee...，问$S_{l..r}$中有几个字母跟这个新的字符串一一对应。

SOL：对于每个字母，用BIT[x][y][L]表示$S_{1..L}$中，所有$equiv xleft (mod \, y ight )$的位置出现了该字母几次。然后复杂度大概就是$O(100*mlogn)$。

然而比赛的时候由于没有想到树状数组动态更新前缀和，而是用了一个静态数组，强行用定期重构过去了。。复杂度$O(10 * m*sqrt{n})$，略微有些暴力。

#include <set>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int LEN = 1e5 + 5;
int i, j, k, n, m, s, t, ans, S, top;
char c[LEN], a[LEN], d[LEN];
int f[LEN][15][4];
set <int> st;
typedef set <int> :: iterator iter;
const char to[4] = {'A', 'C', 'G', 'T'};
int get(char x) {
    if (x == 'A') {
        return 0;
    } else if (x == 'C') {
        return 1;
    } else if (x == 'G') {
        return 2;
    } else {
        return 3;
    }
}
void init() {
    for (int i = n; i >= 1; i--) {
        for (int j = 1; j <= 10; j++) {
            for (int k = 0; k < 4; k++) {
                if (get(c[i]) == k) {
                    f[i][j][k] = 1;
                } else {
                    f[i][j][k] = 0;
                }
                if (i + j <= n) {
                    f[i][j][k] += f[i + j][j][k];
                }
            }
        }
    }
}
int ask(int l, int r, int L) {
    int ans = 0;
    if (r - l + 1 <= L) {
        for (int i = l; i <= r; i++) {
            if (c[i] == a[i - l + 1]) {
                ans++;
            }
        }
    } else {
        for (int i = 1; i <= L; i++) {
            int t = get(a[i]);
            ans += f[l + i - 1][L][t];
            int k = (r - l + 1 - i) / L + 1;
            if (k >= 0 && l + i - 1 + k * L <= n) {
                ans -= f[l + i - 1 + k * L][L][t];
            }
        }
    }
    for (iter it = st.begin(); it != st.end(); it++) {
        int x = *it, t = get(d[x]);
        if (l <= x && x <= r) {
            int p = (x - l) % L + 1;
            if (get(c[x]) == get(a[p]) && get(a[p]) != t) {
                ans--;
            }
            if (get(c[x]) != get(a[p]) && get(a[p]) == t) {
                ans++;
            }
        }
    }
    return ans;
}
void rebuild() {
    for (iter p = st.begin(); p != st.end(); p++) {
        int x = *p;
        c[x] = d[x];
    }
    init();
    st.clear();
}
int main() {
    scanf("%s", c + 1);
    n =  strlen(c + 1);
    S = sqrt(n) + 200;
    init();
    scanf("%d", &m);
    while (m--) {
        int op, l, r;
        scanf("%d", &op);
        if (op == 2) {
            scanf("%d %d", &l, &r);
            scanf("%s", a + 1);
            int L = strlen(a + 1);
            printf("%d
", ask(l, r, L));
        } else {
            scanf("%d %s", &l, a + 1);
            st.insert(l);
            d[l] = a[1];
            if (st.size() >= S) {
                rebuild();
            }
        }
    }
    return 0;
}