CF1207F Remainder Problem

CF1207F. Remainder Problem 暴力分治

Notes

给定一个50万大小的一维数组，有以下两个操作，进行最多50万次询问

• 1 $x$ $y$ 把a[x]加上$y$

• 2 $x$ $y$ 求在$a$数组中所有$imodx=y$的a[i]的和
  
  复杂度证明

• 假设以$B$为界，预处理出模数小于$B$的答案，查询时如果模数小于$B$就直接返回。预处理$B$矩阵所用时间为$O(B)$

• 模数大于$B$，暴力计算所用时间为$O(\frac{N}{B})\to 求每个数消耗的时间，这里不要省去B$

每个查询的复杂度可以表示为
$$O(\frac{N}{B}+B)\to O_{min}=Min(\frac{N}{B}+B)⟺\frac{N}{B}+B\ge \sqrt{\frac{N}{B}\times B}=\sqrt{N}(\frac{N}{B}=B时)$$
所以总复杂度为

$$O(q\times \sqrt{N})$$

Code

#include <bits/stdc++.h>

using namespace std;
#define ll long long
#define mod 1000000007
const ll maxn = 2e6 + 7;
int a[600000];
int ans[800][800];//记录所有k=(1~5e5)中所有a[k]%i==j的和

int main() {
	ios::sync_with_stdio(false);
    int q, t, x, y;
    cin >> q;
    while (q--) {
        cin >> t >> x >> y;
        if (t == 1) {
            a[x] += y;
            for (long long i = 1; i <= 750; ++i) {
                ans[i][x % i] += y;
            }

        } else if (t == 2) {
            if (x <= 750) cout << ans[x][y] << '\n';
            else {
                ll tmp = 0;
                for (long long i = y; i <= 500000; i += x) {
                    tmp += a[i];
                }
                cout << tmp << '\n';
            }
        }
    }
    return 0;
}