假定有一个无限长的数轴,数轴上每个坐标上的数都是 0。
现在,我们首先进行 n
次操作,每次操作将某一位置 x
上的数加 c
。
接下来,进行 m
次询问,每个询问包含两个整数l
和r
,你需要求出在区间[l,r]
之间的所有数的和。
输入格式
第一行包含两个整数n
和m
。
接下来n
行,每行包含两个整数 x
和c
。
再接下来 m
行,每行包含两个整数 l
和 r
。
输出格式
共 m
行,每行输出一个询问中所求的区间内数字和。
数据范围
−10e9≤x
≤10e9,
1≤n,m
≤105,
−10e9≤l
≤r
≤10e9,
−10000≤c
≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
思路
维护的区间为[-10e9,10e9],而用到的x和2m,实际只会用到3x10e5个下标,数据是稀疏的,所以会有离散化的思路。将数据映射到以自然数1,2,3…n的下标的心数组中存储。
代码解析
- 对于每次询问共有2m个下标,加上n个x,所以我们设置N = 300010
- 因为是成对的操作,所以使用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) {//求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开始的数组中,方便求前缀和
}
//删除数组中重复数据,然后将不重复数据放在前面,返回不重复数的最后一个数的下标
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];
}
}
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 });//x + c
alls.push_back(x);//x添加到待离散化的数组alls里
}
for (int i = 0; i < m; i++) {
int l, r;
cin >> l >> r;
query.push_back({ l,r });//左右区间加到query中
alls.push_back(l); //左右区间加到待离散化的数组alls中
alls.push_back(r);
}
//待离散化的数组alls去重,1先排序,2再去重
sort(alls.begin(), alls.end());
//unique函数将重复的元素删去,再将不重复的数依次排过去,并返回最后一个元素的下标
//然后删除掉不重复元素的最后一个下标和区间最后的下标之间的冗余元素
alls.erase(unique(alls), alls.end());
//处理插入,add方法
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;
}
参考
1.https://www.acwing.com/problem/content/804/