AcWing802 区间和 离散化 题解

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int, int> PII;
const int N = 300010;
int a[N], s[N];
int n, m;
vector<int>alls;//存储x,l,r;
vector<PII>query, add;//query存储整组l,r add存储整组x,c
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;//返回>=x的第一个位置
		else l = mid + 1;
	}
	return r + 1;
}
int main() {
	s[0] = 0;
	cin >> n >> m;//n次操作与m行询问
	for (int i = 1; i <= n; i++) {
		int x, c;
		cin >> x >> c;
		add.push_back({ x,c });
		alls.push_back(x);
	}
	for (int i = 1; i <= m; i++) {
		int l, r;
		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 x = find(item.first);//返回add里的x所对应的离散化后的下标 把c可以放到对应位置
		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);
		cout << s[r] - s[l - 1] << endl;
	}
	return 0;
}