acwing算法基础课：离散化

离散化模板

vector<int> alls; // 存储所有待离散化的值
sort(alls.begin(), alls.end()); // 将所有值排序
alls.erase(unique(alls.begin(), alls.end()), alls.end());   // 去掉重复元素

// 二分求出x对应的离散化的值
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; // 映射到1, 2, ...n
}

例题

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

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

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

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

int main()
{
	cin >> n >> m;
	
	for(int i = 0; i < n; i++)
	{
		int x, c;
		cin >> x >> c;
		
		add.push_back({x, c});
		
		alls.push_back(x);
	}	
	
	for(int i = 0; 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());
			
	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);
		cout << s[r] - s[l - 1] << endl;
	}
	
	return 0;
}

测试样例

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

输出样例
8
0
5