题目:
题解:
区间和,使用离散化的原因为存储的下标过大,如果直接开这么大的数组,根本不现实,第二个原因,本文是数轴,要是采用下标的话,可能存在负值,所以也不能,这个题难点就在于映射关系;
思路:
- 将分散的值映射到连续的数组内
Pair对象add内存储的是输入的{x,c}值;x表示坐标值,c表示加上的数
alls数组内存储的是从add数组内择出来的坐标值,此时的数组内值是无序混乱的,需要对它进行排序去重;
a数组内存储最终的c值,根据alls内的x值,从add数组内找到c值存入a数组;最后a数组内的c值与alls数组一一对应;
最终寻找区间[l,r]内的数值和就直接对a进行前缀和操作了;前缀和操作需要有差分数组,因此需要再构造一个s[]数组用于存储差分值;
因为前缀和操作是需要找到l和r坐标,因此在alls内也应该存入进去
Find()函数是用来返回该数在alls内的下标值的
分析:
例如:输入的{x,c}分别为{1,2}{3,6}{7,5} 查找的{l,r}为{4,7}
add[0]={1,2} add[1]={3,6} add[2]={7,5}
此时下标有1,3,7; l,r的4,7也需要存入我们把它存到alls数组内
alls数组内的值为{1,3,7,4,7} ;alls数组内存入的下标值所以要排序去重变为{1,3,4,7}
alls[0]=1,alls[1]=3,alls[2]=4,all[3]=7;
此时,我们查找add内
当x=1时,c值为2,存入a[0]=2
当x=3时,c值为6,存入a[1]=6;
当x=4时,c值为0,存入a[2]=0;
当x=7时,c值为5,存入a[3]=5;
此时alls与a内的数值就一一对应了;
需要查找区间{4,7}内的数值和,就只要在alls数组内找4与7对应的下标为2,3,然后去a数组内计算区间[2,3]内的前缀和即可;
代码:
#include<iostream>
#include<vector>
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){//find查找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的原因是为了防止r=0,因为前缀和操作中都是从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.push_back(x);//将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());
//处理插入
for(auto item:add){//遍历add数组
int x=find(item.first);//add数组内的第一个值,即x值
a[x]+=item.second;//定位到x值对应的c值,将c值插入到a数组内
}
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;
}