AcWing 802. 区间和
原题链接
简单
作者:
qiao
,
2022-03-10 15:08:09
,
所有人可见
,
阅读 130
C++ 代码
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int N = 300010;//需要离散化的有:x,l,r总数为3e5
int n, m;
int a[N], s[N];//a[]是离散化后的数组
vector<int> alls;//存储所以需要离散化的下标(注意:离散化是对下标的离散化,而不是对下标对应值的离散化)
vector<PII> add, query;
//返回下标x离散化后对应的值
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;//r+1是为了让下标从1开始(便于处理前缀和)
}
int main()
{
cin >> n >> m;
while (n--)
{
int x, c;scanf("%d%d", &x, &c);
add.push_back({ x,c });
alls.push_back(x);
}
while (m--)
{
int l, r; scanf("%d%d", &l, &r);
query.push_back({ l,r });
//在查询[l,r]的和时,我们使用的并不是原始下标,而是离散化后的下标,因此需要对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);
a[x] += item.second;
}
for (int i = 1; i <= alls.size(); i++)s[i] = a[i] + s[i - 1];
for (auto item : query)
{
//注意:我们在使用的时候,均使用的为离散化后的值
int l = find(item.first), r = find(item.second);
printf("%d\n", s[r] - s[l - 1]);
}
}
/*
核心思想:
(1)将原下标进行映射(x->alls)
(2)将映射后的值作为下标取值(alls->a)
*/