例题(区间和)
假定有一个无限长的数轴,数轴上每个坐标上的数都是0。
现在,我们首先进行 n 次操作,每次操作将某一位置x上的数加c。
接下来,进行 m 次询问,每个询问包含两个整数l和r,你需要求出在区间[l, r]之间的所有数的和。
输入格式
第一行包含两个整数n和m。
接下来 n 行,每行包含两个整数x和c。
再接下里 m 行,每行包含两个整数l和r。
输出格式
共m行,每行输出一个询问中所求的区间内数字和。
数据范围
−109≤x≤109,
1≤n,m≤105,
−109≤l≤r≤109,
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
思路
对于该题,数轴x的范围为-1099~109,而对于插入的n次操作,n的范围仅为105,相当于在很大一块地方,数轴上的值为0,对于九次方的数组进行遍历,无疑会TLE。
对于这种值域很大,而用到的地方很小的数组,我们采用离散化操作,即把插入之后的相对位置的大小,当做原本的下标进行操作。
故我们需要开一个一维数组储存需要离散化的值(题中出现的各个底标),该数组大小至少为306。
源代码
❤️C++代码如下❤️
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int,int> PII;
const int N=3e6+10;
int a[N],s[N];//a数组存数据,s为前缀和
vector<int> alls;//存储需要离散化的值
vector<PII> add,query;
int find(int x);
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;
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());
//unique删除重复元素
//接着删除新数组末尾到原数组末尾中间的数组
//处理插入
for(auto item:add){
int x=find(item.first);
a[x]+=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);
cout<<s[r]-s[l-1]<<endl;
}
return 0;
}
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开始
}
例题(区间合并)
给定 n 个区间 [li,ri],要求合并所有有交集的区间。
注意如果在端点处相交,也算有交集。
输出合并完成后的区间个数。
例如:[1,3]和[2,6]可以合并为一个区间[1,6]。
输入格式
第一行包含整数n。
接下来n行,每行包含两个整数 l 和 r。
输出格式
共一行,包含一个整数,表示合并区间完成后的区间个数。
数据范围
1≤n≤100000,
−109 ≤li≤ri≤ 109
输入样例:
5
1 2
2 4
5 6
7 8
7 9
输出样例:
3
思路
将每次需要求的区间左右边界存入二维数组,进行排序后,依次进行遍历,判断目前区间右边界与下一区间左边界的大小比较,若右边界大于等于下一区间的左坐标,则判断大小并更新该区间右边界。否则存下该区间,判断下一区间。
源代码
❤️C++代码如下❤️
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int,int> PII;
const int N=1e6+10;
int n;
vector<PII> segs;
void merge(vector<PII> &segs);
int main(){
cin>>n;
for(int i=0;i<n;i++){
int l,r;
cin>>l>>r;
segs.push_back({l,r});
}
merge(segs);
cout<<segs.size()<<endl;
return 0;
}
void merge(vector<PII> &segs)
{
vector<PII> res;
sort(segs.begin(),segs.end());//优先排first,再排second
int st=-2e9,ed=-2e9;
for(auto seg:segs){
if(ed<seg.first){
if(st!=-2e9) res.push_back({st,ed});
st=seg.first;
ed=seg.second;
}
else ed=max(ed,seg.second);
}
if(st!=-2e9) res.push_back({st,ed});
segs=res;
}