洛谷P3396 哈希冲突

分块还真是应用广泛啊......

题意：求

解：以n^0.5为界。

当p小于n^0.5的时候，直接用p²大小的数组储存答案。

预处理n^1.5，修改n^0.5。

当p大于n^0.5的时候，直接按照定义计算，复杂度n^0.5。

所以总复杂度n^1.5，实在是巧妙不堪啊......(什么SB词汇)

#include <cstdio>
#include <algorithm>
#include <cmath>

const int N = 150010;

int fr[N], le[N], re[N];
int ans[400][400], a[N];
char str[3];

int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    int T = sqrt(n);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        fr[i] = (i - 1) / T + 1;
    }
    for(int i = 1; i <= fr[n]; i++) {
        le[i] = re[i - 1] + 1;
        re[i] = le[i] + T - 1;
        if(i == fr[n]) {
            re[i] = n;
        }
    }
    for(int i = 1; i <= T; i++) {
        for(int j = 1; j <= n; j++) {
            ans[i][j % i] += a[j];
        }
    }

    for(int i = 1, x, y; i <= m; i++) {
        scanf("%s%d%d", str, &x, &y);
        if(str[0] == 'A') { // ask
            if(x <= T) {
                printf("%d\n", ans[x][y]);
            }
            else {
                int ans = 0;
                for(int k = 0; k * x + y <= n; k++) {
                    ans += a[k * x + y];
                }
                printf("%d\n", ans);
            }
        }
        else { // change
            for(int j = 1; j <= T; j++) {
                ans[j][x % j] += y - a[x];
            }
            a[x] = y;
        }
    }

    return 0;
}