802. 区间和

本题练习离散化，二分找位置，数组去重。

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 100010;

struct node{
    int x, c;
}nodes[N];

int n, m, k = 1;
int s[N];

int cmp(const node &a, const node &b){
    return a.x < b.x;
}

int findl(int x){ // 找最左边的满足大于等于x的位置
    int l = 1, r = k;
    while(l < r){
        int mid = l + r >> 1;
        if(nodes[mid].x >= x) r = mid;
        else l = mid + 1;
    }
    
    if(nodes[l].x < x) return k + 1; // 如果找到的最左边的位置小于x，说明数组中的x全部小于x，返回k + 1, 因为要用s[r] - s[l - 1]
    return l;
}

int findr(int x){// 找最右边的满足小于等于x的位置
    int l = 1, r = k;
    while(l < r){
        int mid = l + r + 1 >> 1;
        if(nodes[mid].x <= x) l = mid;
        else r = mid - 1;
    }
    
    if(nodes[l].x > x) return 0; // 如果找到的最右边的位置大于x，说明所有x都大于x，返回0。
    return l;
}

int main(){
    cin >> n >> m;
    
    for(int i = 1; i <= n; i ++) cin >> nodes[i].x >> nodes[i].c;
    
    //升序排序
    sort(nodes + 1, nodes + n + 1, cmp);

    //去重
    for(int i = 2; i <= n; i ++)
        if(nodes[i].x == nodes[k].x) nodes[k].c += nodes[i].c;
        else nodes[++ k] = nodes[i];
        
    //预处理前缀和
    for(int i = 1; i <= n; i ++) s[i] = s[i - 1] + nodes[i].c;
    
    while(m --){
        int a, b;
        cin >> a >> b;

        //找左右区间
        int l = findl(a), r = findr(b);
        
        cout << s[r] - s[l - 1] << endl;
    }
    
    return 0;
}