Codeforces 1207F Remainder Problem

tags

根号分治

中文题面

给你一个由 $500000$ 个整数（编号从 $1$ 到 $500000$ ）组成的数组 $a$ 。最初 $a$ 中的所有元素都为零。

您必须对这个数组进行两种类型的查询：

• $1$ $x$ $y$ - $a_x$ 增加 $y$ ；
• $2$ $x$ $y$ - 计算 $\sum _{i\in R(x,y)}{a}_i$ ，其中 $R(x,y)$ 是所有从 $1$ 到 $500000$ 有余数 $y$ modulo $x$ 的整数的集合。

你能处理所有的查询吗？

输入

第一行包含一个整数 $q$ ( $1\le q\le 500000$ ) - 查询次数。

然后是 $q$ 行，每行描述一个查询。第 $i$ 行包含三个整数 $t_i$ 、 $x_i$ 和 $y_i$ （ $1\le t_i\le 2$ ）。如果是 $t_i=1$ ，则是第一类查询、 $1\le x_i\le 500000$ 和 $-1000\le y_i\le 1000$ 。如果是 $t_i=2$ ，则是第二类查询、 $1\le x_i\le 500000$。

可以保证至少有一个查询属于 $2$ 类型。

输出

为每个 $2$ 类型的查询打印一个整数–答案。

思路

没有预处理，直接暴力，时间复杂度为O(q * n / x); 预处理时间复杂度O(nx)，每次O(1)回答，O(x)修改，总时间复杂度O(nx)，使用根号分治，当x <= 707时使用预处理方式解决，当x > 707时使用暴力算法解决，避免超时，本题得解

代码

#include <bits/stdc++.h>
using namespace std;

int s[708][708];
int v[500005];
// 707分界
inline void solve() {
    const int n = 500000;
    int m, x, y;
    char op;
    cin >> m;
    while (m--) {
        cin >> op >> x >> y;
        if (op == '2') {
            if (x <= 707) cout << s[x][y] << '\n';
            else {
                int res = 0;
                for (register int i = 0; i * x + y <= n; i++) res += v[i * x + y];
                cout << res << '\n';
            }
        }
        else {
            v[x] += y;
            for (register int p = 1; p <= 707; p++) s[p][x % p] += y;
        }
    }
}
int main() {
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    solve();
    return 0;
}