C++ / ACM 自学笔记 · v5.0

P8218 求区间和

原题入口

对应章节:序言 → 建议的自学顺序与练习;五、数据结构与算法 → (三) 前缀和与差分。

(一) 任务与输出目标

回答多次闭区间求和询问。

(二) 建模与推导

S[i] 保存前 i 项之和,S[0]=0。前 r 项去掉前 l-1 项,剩下的恰好是 l 到 r,因此答案为 S[r]-S[l-1]。预处理时每项加一次,查询时只做一次减法。

(三) 手算与过程

1,2,3,4 的前缀为 0,1,3,6,10;查询 [2,3] 得 6-1=5。

按具体数据展开的处理与返回过程

(四) 带注释实现

完整源文件

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;

void solve() {
    int n;
    cin >> n;
    vector<int> sum(n + 1, 0);
    for (int i = 1; i <= n; ++i) {
        int x;
        cin >> x;
        sum[i] = sum[i - 1] + x;
    }
    int q;
    cin >> q;
    while (q--) {
        int l, r;
        cin >> l >> r;
        cout << sum[r] - sum[l - 1] << '\n'; // 去掉左端点之前的前缀。
    }
}

signed main() {
    ios::sync_with_stdio(false); // 关闭同步,使用 cin 与 cout 完成输入输出。
    cin.tie(0), cout.tie(0);

    int T = 1; // 本题读入一组数据。
    // cin >> T; // 题目给出测试组数时开启,并在 solve 中处理一组。
    while (T--)
        solve();
}

(五) 复杂度

时间 \(O(n+q)\),空间 \(O(n)\)。

返回本章题解索引 · 返回全部题号索引

查阅笔记正文与全部题解 · Ctrl / ⌘ K