题目传送门:802. 区间和
假定有一个无限长的数轴,数轴上每个坐标上的数都是 0。
现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c。
接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r] 之间的所有数的和。
输入格式
第一行包含两个整数 n 和 m。
接下来 n 行,每行包含两个整数 x 和 c。
再接下来 m 行,每行包含两个整数 l 和 r。
输出格式
共 m 行,每行输出一个询问中所求的区间内数字和。
数据范围
−1e9≤x≤1e9,
1≤n,m≤105,
−1e9≤l≤r≤1e9,
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
试题解析:
本题x范围为-1e9~1e9,值域跨度很大,但是数据稀疏,需要用到离散化知识,即用映射的方式,将间隔很大的点,映射到相邻的数组元素中。
- 创建数组来存储坐标插入的值,开辟空间考虑为300000+,因为存在n+2m次查询。
- 本题使用alls数组存储n次插入中的每个x,还存储m次查询的每次的l与r。
- 重新排列去重数组,利用find函数找到大于等于所填数的最小的下标,进行插入操作,返回的l+1是为了接下来的前缀和数组,使能插入数据的数组的最小的下标为1。
- 进行前缀和操作,通过find函数处理后m次询问操作,利用前缀和的方法求出最后的区间和。
附图讲解:
注:此图解来源于acwing中liangshang大佬对于本题的讲解。
代码如下:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N = 300010;//n+2m
int n, m;
typedef pair<int, int> PII;
vector<int> alls;//存储所有坐标
vector<PII> add, query;//存储插入和询问操作的数据
int a[N];//存储坐标插入的值
int s[N];//存储数组a的前缀和
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() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
int x, c;
scanf("%d%d", &x, &c);
add.push_back({x, c});
alls.push_back(x);
}
for (int i = 1; i <= m; i ++) {
int l, r;
scanf("%d%d", &l, &r);
alls.push_back(l);
alls.push_back(r);
query.push_back({l, r});
}
//排序去重
sort(alls.begin(), alls.end());
alls.erase(unique(alls.begin(), alls.end()), alls.end());
//进行前n次插入操作
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];
}
//处理后m次询问操作
for (auto item : query) {
int l = find(item.first);
int r = find(item.second);
printf("%d\n", s[r] - s[l - 1]);
}
return 0;
}