离散化模板与例题

情景： 给定的一个范围很大的区间，但是实际上用到的很小，较为稀疏，若采用暴力枚举时间复杂度会很高，而经过离散化后只对有效数据进行处理，再次枚举时就会大大缩短枚举区间，降低时间复杂度

模板：

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
}

---

---

acwing802. 区间和

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 3e5 + 10;
typedef pair<int, int> PII;
int n, m;
int a[N], s[N];

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

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 l + 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 i : add)
    {
        int x = find(i.first);
        a[x] += i.second;		//将离散化后的下标对应到他的值中
    }
    
    for(int i = 1; i <= alls.size(); i ++ ) s[i] = s[i - 1] + a[i];		//前缀和预处理
    
    for(auto i : query)
    {
        int l = find(i.first), r = find(i.second);
        cout << s[r] - s[l - 1] << endl;
    }
    return 0;
}