区间和(离散化)
假定有一个无限长的数轴,数轴上每个坐标上的数都是0。
现在,我们首先进行 n 次操作,每次操作将某一位置x上的数加c。
接下来,进行 m 次询问,每个询问包含两个整数l和r,你需要求出在区间[l, r]之间的所有数的和。
输入格式
第一行包含两个整数n和m。
接下来 n 行,每行包含两个整数x和c。
再接下里 m 行,每行包含两个整数l和r。
输出格式
共m行,每行输出一个询问中所求的区间内数字和。
数据范围
−
1
0
9
−10^9
−109≤x≤
1
0
9
10^9
109,
1≤n,m≤105,
−
1
0
9
−10^9
−109≤l≤r≤
1
0
9
10^9
109,
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N=300000;
typedef pair<int,int> PII;
vector<int> alls;
vector<PII> add,query;
int a[N],s[N];
int find(int x)
{
int l=0,r=alls.size()-1;
while(r>l)
{
int mid=(l+r)/2;
if(alls[mid]>=x) r=mid;
else l=mid+1;
}
return l+1;
}
int main()
{
int n,m;
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);
}
for(int i=0;i<m;i++)
{
int l,r;
cin>>l>>r;
alls.push_back(l);
alls.push_back(r);
query.push_back({l,r});
}
sort(alls.begin(),alls.end());
alls.erase(unique(alls.begin(),alls.end()),alls.end());
for(auto item:add)
{
int t=find(item.first);
a[t]+=item.second;
}
for(int i=1;i<=alls.size();i++) s[i]=s[i-1]+a[i];
for(auto item:query)
{
int l=find(item.first);
int r=find(item.second);
cout<<s[r]-s[l-1];
cout<<endl;
}
}
首先,由于题目中设计到操作的范围比较大而实际操作的范围用不到那么多,另外直接用数组的话下标由于有负值进而无法实现。于是将x,l,r也就是用到的数组的下标进行离散化操作,通过映射实现将下标设置为紧凑而且合理的值。
用vector,int存alls,用vector,pair存add,query。
然后用前缀和处理。