AcWing 802. 区间和
原题链接
简单
作者:
._5144
,
2023-11-16 21:45:14
,
所有人可见
,
阅读 49
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 300010;
int a[N],s[N];
int n,m;
vector<int> alls;
vector<pair<int,int>> 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;
}
int main()
{
cin >> 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); // 把要离散化的数据存入alls
}
// 读入查询区间的左右端点
for(int i = 0; i < m; i ++)
{
int l,r;
scanf("%d%d",&l,&r);
query.push_back({l,r});
// 区间左右端点也需要离散化,故也要存入alls
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);
a[x] += item.second;
}
// 构造前缀和数组
for(int i = 1; i <= alls.size(); i ++) s[i] = s[i - 1] + a[i];
// 计算[l,r]区间的和
for(auto item : query)
{
int l = find(item.first);
int r = find(item.second);
printf("%d\n",s[r] - s[l - 1]);
}
return 0;
}