假定有一个无限长的数轴,数轴上每个坐标上的数都是 00。
现在,我们首先进行 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
解题思路
1,把所有需要赋值的元素下标和需要查询区间的下标都存入一个动态数组中(alls)
for(int i=0;i<n;i++){
int x,c;
cin>>x>>c;
add.push_back({x,c}); //先存着留着后面alls排序好后赋值使用
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);
}
2,把所有存入的下标进行排序去重
sort(alls.begin(), alls.end());
alls.erase(unique(alls.begin(), alls.end()), alls.end());
3,利用二分快速的给已经拍好序列的元素赋值
for (auto item : add) {
int x = find(item.first);
a[x] += item.second;
}
//两种代码等价
for(int i=0;i<n;i++){
int x=find(add[i].first);
a[x]+=add[i].second;
}
4,预处理(前缀和)
for (int i = 1; i <= alls.size(); i++) s[i] = s[i-1] + a[i];
5,再遍历一遍需要查询的区间的端点,输出答案。
//query先前存的值现在开始使用,求结果
for(auto item:query){
int l=find(item.first),r=find(item.second);
cout<<s[r]-s[l-1]<<endl;
}
//与上面代码等价
for(int i=0;i<m;i++){
int l=find(query[i].first),r=find(query[i].second);
cout<<s[r]-s[l-1]<<endl;
}
完整代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N=3e5+10;
typedef pair<int,int>PII;
int a[N]; //存储坐标插入的值
int s[N]; //存数组a的前缀和
vector<int>alls; //存储所有的元素去重排序后得到一个新的数组
vector<PII>add,query; //存储插入和询问操作的数据
int n,m;
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;
}
int main(){
cin>>n>>m;
for(int i=0;i<n;i++){
int x,c;
cin>>x>>c;
add.push_back({x,c}); //先存着留着后面alls排序好后赋值使用
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.begin(),alls.end()),alls.end());//去重
//add先前存的值现在开始使用,用原本存的下标来求现在的新下标
//并把存的值加入新的下标元素。
for(auto item:add){
int x=find(item.first);
a[x]+=item.second;
}
//与上面代码等价
//for(int i=0;i<n;i++){
// int x=find(add[i].first);
// a[x]+=add[i].second;
//}
for(int i=1;i<=alls.size();i++) //前缀和操作
s[i]=s[i-1]+a[i];
//query先前存的值现在开始使用,求结果
for(auto item:query){
int l=find(item.first),r=find(item.second);
cout<<s[r]-s[l-1]<<endl;
}
//与上面代码等价
//for(int i=0;i<m;i++){
// int l=find(query[i].first),r=find(query[i].second);
// cout<<s[r]-s[l-1]<<endl;
//}
return 0;
}
视频详解
802. 区间和 - AcWing题库高质量的算法题库https://www.acwing.com/problem/content/video/804/