[AcWing 802] 区间和

---

点击查看代码

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;
typedef pair<int, int> PII;
const int N = 3e5 + 10;
vector<int> alls;
vector<PII> add, query;
int a[N], s[N];

int find(int x)
{
    int l = 0, r = alls.size() - 1;
    while (l < r) {
        int mid = l + r >> 1;
        if (alls[mid] >= x) r = mid;
        else    l = mid + 1;
    }
    return r + 1;
}
int main()
{
    int n, m;
    scanf("%d %d", &n, &m);
    for (int i = 0; i < n; i ++) {
        int x, c;
        scanf("%d %d", &x, &c);
        add.push_back({x, c});
        alls.push_back(x);
    }
    for (int i = 0; i < m; i ++) {
        int l, r;
        scanf("%d %d", &l, &r);
        query.push_back({l, r});
        alls.push_back(l);
        alls.push_back(r);
    }
    sort(alls.begin(), alls.end());
    alls.erase(unique(alls.begin(), alls.end()), alls.end());
    for (auto item : add) {
        int x = find(item.first);
        a[x] += item.second;
    }
    for (int i = 1; i <= alls.size(); i ++) s[i] = s[i - 1] + a[i];
    for (auto item : query) {
        int l = find(item.first), r = find(item.second);
        printf("%d\n", s[r] - s[l - 1]);
    }
    return 0;
}

---

1. pair<int, int> 是将两个 int 变量的元素组成一个数据；
2. const int N = 3e5 + 10 是因为最多有 n + 2m 个元素；
3. alls 用来存所有的需要映射的数，alls 在进行 sort 排序后进行去重，去重的操作：unique(alls.begin(), alls.end() 作用是将所有不重复的元素放到 alls 的前面，重复元素放到后面，返回的迭代器指向重复元素的第一个位置，使用 erase 可以把后面重复的段删除；
4. add 用来存 x 位置加上 c 这个操作；
5. query 用来存 [ l, r ] 区间查询这个操作；
6. 最后使用前缀和进行求和；