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
思路:这类题的重点是如何处理稀疏数据。可以用map对位置和值进行保存,然后结合前缀话的思路即可完成题目。将查询结果同样插入保存map可以很好的处理边界问题。
代码:
#include <iostream>
#include <unordered_map>
#include <unordered_set>
#include <vector>
using namespace std;
typedef pair<int, int> p;
vector<p> alls;
unordered_map<int,int> helper;
vector<p> querry;
int m,n;
void my_quicksort(vector<p>& alls, int l, int r){
if(l>r) return;
int k = alls[r].first;
int i = l-1;
for(int j = l; j < r;j++){
if(alls[j].first<k) swap(alls[++i],alls[j]);
}
swap(alls[i+1],alls[r]);
my_quicksort(alls,l,i);
my_quicksort(alls,i+2,r);
}
int my_binarySearch(vector<p>& alls, int k){
int l = 0;
int r = alls.size()-1;
while(l<r){
int mid = l+r>>1;
if(alls[mid].first>=k) r = mid;
else l = mid+1;
}
return l;
}
int main()
{
cin>>n>>m;
//存放数据进helper进行去重
while (n-- ){
int x,c;
cin>>x>>c;
if(helper.find(x)==helper.end()) helper.insert({x,c});
else helper.find(x)->second+=c;
}
//将Helper的数据放入alls
for(auto p = helper.begin(); p!=helper.end();p++){
alls.push_back({p->first,p->second});
}
querry.push_back({0,0});//处理边界问题
//将待查询数据插入alls 很好的处理边界问题
while (m -- ){
int l,r;
cin>>l>>r;
querry.push_back({l,r});
if(helper.find(l)==helper.end()) alls.push_back({l,0});
if(helper.find(r)==helper.end()) alls.push_back({r,0});
}
//对其alls进行排序
my_quicksort(alls,0,alls.size()-1);
//对all进行前缀和化
for (int i = 1; i < alls.size()+1; i ++ ) alls[i].second = alls[i].second+alls[i-1].second;
//得到结果
for (int i = 1; i < querry.size(); i ++ ){
int l = my_binarySearch(alls,querry[i].first), r = my_binarySearch(alls,querry[i].second);
int res = alls[r].second - alls[l-1].second;
cout<<res<<endl;
}
}