AcWing 802. 区间和(C++)

AcWing 802. 区间和

1、题目（来源于AcWing）：
假定有一个无限长的数轴，数轴上每个坐标上的数都是0。

现在，我们首先进行 n 次操作，每次操作将某一位置x上的数加c。

接下来，进行 m 次询问，每个询问包含两个整数l和r，你需要求出在区间[l, r]之间的所有数的和。

输入格式
第一行包含两个整数n和m。

接下来 n 行，每行包含两个整数x和c。

再接下里 m 行，每行包含两个整数l和r。

输出格式
共m行，每行输出一个询问中所求的区间内数字和。

数据范围
−10⁹≤x≤10⁹,
1≤n,m≤10⁵,
−10⁹≤l≤r≤10⁹,
−10000≤c≤10000

输入样例：
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例：
8
0
5

2、基本思想：
虽然数据范围是−10⁹~10⁹,但是真正用到的下标范围是：
①add：用到的n次操作（10⁵）
②query：用到的m次询问中 l (10⁵) 和 r (10⁵)
一共是300000个数据。

可用离散化将x, l , r映射到1 ~ 300010中。

3、步骤：
①将n次add存入alls；
②将m次query存入alls，l 和 r 分开存；
③将alls离散化；
④用前缀和求区间和。

4、C++代码如下（该代码引用AcWing网站的代码）：

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

using namespace std;

typedef pair<int, int> PII;//pair<int, int>表示两个int变量

int n, m, x, c, l, r;
int a[300010], s[300010];
vector<int> alls;//待离散化的数组
vector<PII> add, query;//插入和询问操作

int find(int x)//二分查找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()
{
    cin >> n >> m;
    for (int i = 0; i < n; i ++ )//将n次x存入alls
    {
        cin >> x >> c;
        add.push_back({x, c});
        alls.push_back(x);
    }
    
    for (int i = 0; i < m; i ++ )//将m次l和r存入alls
    {
        cin >> 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());//alls去重
    
    for (auto item : add)
    {
        int idx = find(item.first);
        a[idx] += item.second;
    }
    
    for (int i = 1; i <= alls.size(); i ++ ) s[i] = s[i - 1] + a[i];//预处理前缀和
    
    for (auto item : query)
    {
        int l, r;
        l = find(item.first);
        r = find(item.second);
        cout << s[r] - s[l - 1] << endl;
    }
    
    return 0;
}//该代码引用AcWing网站的代码

注意事项：
for (auto item : add) 的循环条件就是add的元素个数大于0，item.first访问的是pair型变量add的第一项。