AcWing 802. 区间和
原题链接
简单
作者:
浮玺君
,
2021-02-09 13:07:22
,
所有人可见
,
阅读 276
C++ 代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int , int> PII;
const int N = 3*10e5 + 10;
int n,m;
int a[N],s[N];
vector<int> alls; //存储要离散化的数值
vector<PII> add,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;//前缀和是一般从下标从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); //因为对坐标进行离散化所以加入到alls
}
for(int i = 0; i < m; i++)
{
int l, r;
cin >> l >> r;
add.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)
{
int x = find(item.first); //原来坐标 add.push_back({x,c})中的x 在离散化后的新坐标:x
a[x] += item.second; //在离散化坐标上加上{x,c}中的c 这里item.second == c
}
//预处理前缀和
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);//这里的l,r是add.push_back({l,r})中的l,r在离散化的值
cout << s[r] - s[l-1] << endl; //在[l,r]的区间和
}
return 0;
}