基础算法(十四)离散化+二分 ---以题为例

Ghost-Knight / 2024-02-04 / 原文

假定有一个无限长的数轴,数轴上每个坐标上的数都是 0

现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c

接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r] 之间的所有数的和。

输入格式

第一行包含两个整数 n 和 m

接下来 n行,每行包含两个整数 x 和 c

再接下来 m 行,每行包含两个整数 l 和 r

输出格式

共 m 行,每行输出一个询问中所求的区间内数字和。

数据范围

109x109,
1n,m105,
109lr109,
10000c10000

输入样例:

3 3
1 2
3 6
7 5
1 3
4 6
7 8

输出样例:

8
0
5

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

typedef pair<int,int> PII;

const int N =300010;
int a[N],s[N];

vector<int> alls;
vector<PII> add,query;

int find(int x){
    int l=0;
    int 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(){
    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());
    
    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];
    
    for(auto item:query){
        int l =find(item.first);
        int r =find(item.second);
        cout<<s[r]-s[l-1]<<endl;
    }
    return 0;
}