首页 > 其他 > 详细

802. 区间和

时间:2020-08-30 13:50:04      阅读:51      评论:0      收藏:0      [点我收藏+]

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

#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;
}

802. 区间和

原文:https://www.cnblogs.com/tomori/p/13584665.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!