C++——离散化求区间和

假定有一个无限长的数轴，数轴上每个坐标上的数都是 0。

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

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

输入格式

第一行包含两个整数 n 和 m。

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

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

输出格式

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

数据范围

−109≤x≤1e9
1≤n,m≤1e5
−1e9≤l≤r≤1e9
−10000≤c≤10000

输入样例：

3 3
1 2
3 6
7 5
1 3
4 6
7 8

输出样例：

8
0
5

 解题思路：本题可采用离散化求区间和，即将搜算范围从小到大映射，映射后的坐标从1开始。

Code:

#include <iostream>
#include <stdio.h>
#include <vector>
#include <algorithm>

using namespace std;

typedef pair<int, int> PII;

const int N = 300010;

int n, m, x, c, l, r, b;
int a[N], s[N];

vector<int> alls;
vector<PII> add, query;

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;
}

signed main()
{
	cin >> n >> m;
	while (n -- )
	{
		cin >> x >> c;
		add.push_back({x, c});
		alls.push_back(x);
	}
	while (m -- )
	{
		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());/*删除重复元素*/
	for(auto item : add)
	{
		int b = find(item.first);
		a[b] += 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);
		cout << s[r] - s[l - 1] << endl;
	}
}