原题连接: AcWing 802. 区间和
关键词: 保序离散化模板题、前缀和
假定有一个无限长的数轴,数轴上每个坐标上的数都是 0。
现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c。
接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r] 之间的所有数的和。
输入格式
第一行包含两个整数 n 和 m。
接下来 n 行,每行包含两个整数 x 和 c。
再接下来 m 行,每行包含两个整数 l 和 r。
输出格式
共 m 行,每行输出一个询问中所求的区间内数字和。
数据范围
−109≤x≤109,
1≤n,m≤105,
−109≤l≤r≤109,
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
思路:
和前缀和一样,也是给出n个操作往坐标上加数,m个询问关于[l, r]区间内的数之和。
区别在于他的值的范围很大,但值的个数很少,也就是说它相当 稀疏,直接用前缀和的方法做不好,时间复杂度太高了。
离散化的思路就是,把每个有数的下标,映射成保序的、连续的新下标。在新数组下再求前缀和,又快又好。
例如下面是5个原本的下标:
1 3 77 648 1997 500000
映射为:
1 2 3 4 5
在新数组中求前缀和更好(因为此时全部都是有数的下标,不稀疏了),从1开始是为了方便求前缀和
代码实现:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> PII;
const int maxn = 300010;
int a[maxn], s[maxn]; //a存离散化后的下标的实际值 s是a的前缀和数组
vector<int> v; //存需要进行离散化的数(下标)
vector<PII> add, query; //add存插入操作 query存查询操作
int n, m; //n数的个数 m询问数
// 求x离散化之后的结果 二分
int find(int x){
int l = 0, r = v.size() - 1;
while(l < r){
int mid = l + r >> 1;
if(v[mid] >= x) r = mid;
else l = mid + 1;
}
return r + 1; //因为用到前缀和 新下标采用从1开始
}
int main(){
cin >> n >> m;
for(int i = 0; i < n; i ++ ){
int x, c; //在下标为x的地方加上c
cin >> x >> c;
add.push_back({x, c});
v.push_back(x);
}
for(int i = 0; i < m; i ++ ){
int l, r; //存查询的左右区间
cin >> l >> r;
query.push_back({l, r});
v.push_back(l);
v.push_back(r);
}
// 对v数组排序、去重
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
// 处理插入
for(auto item : add){
int x = find(item.first); //找到下标
a[x] += item.second;
}
// 预处理前缀和
for(int i = 1; i <= v.size(); i ++ )
s[i] = s[i - 1] + a[i];
// 处理询问
for(auto item : query){
int l = find(item.first);
int r = find(item.second);
cout << s[r] - s[l] + a[l] << endl;
}
return 0;
}