整数的离散化:值域比较大(0~10^9),个数比较小(0~10^5),有些题目需要以这些值为下标来做,要把它们映射到从0开始的自然数,这些自然数就是其下标。这个过程就称作离散化
注意两个问题:
1.A[ ]中可能有重复元素,需要去重
//数组排序并去重
vector<int> alls; //存储所有待离散化的值
sort(alls.begin(),alls.end()); //将所有值排序
alls.erase(unique(alls.begin(),alls.end())); //去掉重复元素
- 使用 sort 排序:需要头文件 #include<algorithm>,sort(vec.begin(),vec.end());(默认是按升序排列,即从小到大).
iterator erase(iterator first,iterator last)
:删除向量中[first,last)中元素- unique函数
2.如何算出x离散化后的值(下标):因为A数组是有序的,保序地去离散化,即小的在前面,大的在后面。又因为A数组是有序的,所以可以用二分来进行查找
//二分求出x对应离散化的值
int find(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
}
题目:
数组下标比较小时(10^5以内),前缀和就可以做。值域很大,但是比较稀疏,用离散化做
其中pair用法
代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> PII;
const int N = 300010;
int n, m;
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;
}
vector<int>::iterator unique(vector<int> &a)
{
int j = 0;
for (int i = 0; i < a.size(); i ++ )
if (!i || a[i] != a[i - 1])
a[j ++ ] = a[i];
// a[0] ~ a[j - 1] 所有a中不重复的数
return a.begin() + j;
}
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), 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;
}
有关算法竞赛的题目会继续更新,欢迎评论交流!