先上代码
#include<iostream>
#include <vector>
#include<algorithm>
using namespace std;
const int N=300010;//n次插入和m次查询相关数据量的上界
int n,m;
int a[N];//初始化都为0,存储坐标插入的值
int s[N];//数组a的前缀和
vector<int> alls;//存储所有插入和查询有关的坐标
typedef pair<int,int> PII;
vector<PII> add;//存储插入的数据
vector<PII> query;//存储访问操作的数据
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()
{
scanf("%d%d", &n, &m);
for(int i=0;i<n;i++)
{
int x,c;
scanf("%d%d",&x,&c);
add.push_back({x,c});
alls.push_back(x);
}
for(int i=0;i<m;i++)
{
int l,r;
scanf("%d%d",&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());
//执行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;
}
为什么alls要存储l与r呢?
答:我们最终要对[l, r]求和,自然就不能使用原来的数组求和(数据太大,过于离散)。
所以我们把l和r也映射到alls数组(l,r的值若没有add的操作,即插入不会在l或r的位置,值就为0(全局变量的默认值为0))。这样数轴就会变短很多(最大到3*10^5),我们就可以使用前缀和求和达到O(1)。
要注意C++算法里面的排序去重操作
//排序,去重
sort(alls.begin(),alls.end());
alls.erase(unique(alls.begin(),alls.end()),alls.end());
借鉴一下别人的图