离散化
① a中有重复元素 => 去重
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();
while(l < r)
{
int mid = l + r >> 1;
if(alls[mid] >= x) r = mid;
else l = mid + 1;
}
return r + 1; //离散 从1开始
}
例题:
采用离散化的方式,离散坐标。
alls
:将所有需要离散的坐标存入该数组
add
:存放插入的位置及数字(插入的位置需要进行离散化 => 存入alls
)
query
:存放询问的区间(起始位置和终止位置需要进行离散化 => 存入alls
)
a
:存放离散后的坐标对应的值。
s
:存放a数组的前缀和(用于求区间和)
具体处理:
-
处理插入时:将离散后的坐标对应a数组,存入相应的值。
-
处理查询时:求离散后的区间的前缀和。
s[r] - s[l-1]
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 300010; //插入需要一个坐标,查询需要两个坐标 n+2m
typedef pair<int, int> PII;
vector<PII> add; //插入位置及值
vector<PII> query; //查询的两个坐标
vector<int> alls; //存放需要待离散化的坐标
int a[N],s[N]; //原数组、前缀和数组
//通过二分返回离散化后的值
int find(int x) //寻找第一个大于等于x的位置
{
int l = 0, r = alls.size();
while(l < r)
{
int mid = l + r >> 1;
if(alls[mid] >= x) r = mid;
else l = mid + 1;
}
return r + 1; //离散 从1开始
}
int main()
{
int n,m;
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); //待处理坐标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]; //a[]是从1开始记录的,需要遍历的大小为alls的长度
for(auto item : query)
{
int l = find(item.first);
int r = find(item.second);
cout << s[r] - s[l-1] << endl;
}
return 0;
}