Time to Raid Cowavans 分块，离线

Time to Raid Cowavans

题意

一个长度为 $n$ 的数组 a ，给出 $q$ 个询问，询问给出两个值 $pos,x$ ，要求给出 $$a_{pos}+a_{pos+x}+a_{pos+2\cdot x}+\dots$$ 的值。

题解

• $B=\sqrt{n}$ ，如果 $x\ge B$ ，那么直接暴力即可，复杂度 $O(\frac{n}{x})<O(\sqrt{n})$ ；
• 那么剩下不同的 $x$ 小于 $B$ 个，离线询问，按照 $x$ 排序，对相同的 $x$ ，处理出 $sum[i]$ 表示 $pos=i$ 时的答案，处理的复杂度为 $O(n)$ 。
• 总复杂度为 $O(n\cdot \sqrt{n})$ 。

代码

#include <bits/stdc++.h>
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#ifdef LOCAL
#include "Print.h"
#define de(...) W('[', #__VA_ARGS__,"] =", __VA_ARGS__)
#else
#define de(...)
#endif
using namespace std;
typedef long long ll;
const int maxn = 3e5 + 5;
int n, q, B;
ll a[maxn], ANS[maxn];
struct query {
    int pos, x, id;
} Q[maxn];
int solve() {
    scanf("%d", &n);
    B = sqrt(n);
    rep(i, 1, n) scanf("%lld", &a[i]);
    scanf("%d", &q);
    auto cul1 = [&](int pos, int x) {
        ll ans = 0;
        for (int i = pos; i <= n; i += x) ans += a[i];
        return ans;
    };
    vector<query> v;
    rep(i, 1, q) {
        scanf("%d%d", &Q[i].pos, &Q[i].x), Q[i].id = i;
        if (Q[i].x >= B) ANS[i] = cul1(Q[i].pos, Q[i].x);
        else v.push_back(Q[i]);
    }
    sort(v.begin(), v.end(), [](const query &A, const query &B) {
        return A.x < B.x;
    });
    vector<ll> sum(n * 2 + 5); // 以i为pos计算答案
    for (int i = 0; i < v.size(); ++i) {
        auto cul2 = [&](int x) {
            for (int i = n; i >= 1; --i)
                sum[i] = sum[i + x] + a[i];
        };
        if (!i || v[i].x != v[i - 1].x) cul2(v[i].x);
        ANS[v[i].id] = sum[v[i].pos];
    }
    rep(i, 1, q) printf("%lld\n", ANS[i]);
    return 0;
}
int main() {
#ifdef LOCAL
    freopen("in.in", "r", stdin);
    freopen("out.out", "w", stdout);
#endif
    int T = 1;
    // scanf("%d", &T);
    while (T--) solve();
    return 0;
}