ABC 288 D - Range Add Query

题目链接：

设 \(a[i]\) 表示下标对 \(k\) 取模为 \(i\) 的所有数的和。那每次操作就是将数组 \(a\) 的所有数都加 \(c\)。最终为 \(0\) 的充分必要条件就是 \(a\) 的所有数都是一样的。

因此，对于每组询问，统计 \([l,r]\) 中数字下标对 \(k\) 取模的所有数的和，看看是否为同一个数即可。也即判断下标 \(i\) % \(k\) 的 \(k\) 个前缀和是否相同。预处理原数组的下标取模前缀和，每组询问就两个前缀和相减就得到该区间的下标取模前缀和。

#include <bits/stdc++.h>

using i64 = long long;

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    int n, k;
    std::cin >> n >> k;
    
    std::vector<int> a(n);
    std::vector<i64> s(n);
    for (int i = 0; i < n; i++) {
        std::cin >> a[i];
        s[i] = a[i];
        if (i >= k) {
            s[i] += s[i - k];
        }
    }
    
    int q;
    std::cin >> q;
    
    while (q--) {
        int l, r;
        std::cin >> l >> r;
        l--, r--;
        
        std::vector<i64> b(k);
        for (int i = r; r - i + 1 <= k; i--) {
            b[i % k] += s[i];
        }
        for (int i = l - 1; i >= 0 && l - i <= k; i--) {
            b[i % k] -= s[i];
        }
        std::vector<i64> c(k, b[0]);
        if (b == c) {
            std::cout << "Yes\n";
        } else {
            std::cout << "No\n";
        }
    }
    
    return 0;
}