区间和 : 离散化 + 前缀和

原题连接: AcWing 802. 区间和
关键词: 保序离散化模板题、前缀和

假定有一个无限长的数轴,数轴上每个坐标上的数都是 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

思路:
和前缀和一样,也是给出n个操作往坐标上加数,m个询问关于[l, r]区间内的数之和。
区别在于他的值的范围很大,但值的个数很少,也就是说它相当 稀疏,直接用前缀和的方法做不好,时间复杂度太高了。

离散化的思路就是,把每个有数的下标,映射成保序的、连续的新下标。在新数组下再求前缀和,又快又好。
例如下面是5个原本的下标:

1 3 77 648 1997 500000 

映射为:

1 2 3 4 5

在新数组中求前缀和更好(因为此时全部都是有数的下标,不稀疏了),从1开始是为了方便求前缀和


代码实现:

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

typedef pair<int, int> PII;

const int maxn = 300010;
int a[maxn], s[maxn];   //a存离散化后的下标的实际值 s是a的前缀和数组
vector<int> v;  //存需要进行离散化的数(下标)
vector<PII> add, query; //add存插入操作 query存查询操作

int n, m;   //n数的个数 m询问数

// 求x离散化之后的结果 二分
int find(int x){
    int l = 0, r = v.size() - 1;
    while(l < r){
        int mid = l + r >> 1;
        if(v[mid] >= x) r = mid;
        else l = mid + 1;
    }
    return r + 1;   //因为用到前缀和 新下标采用从1开始
}

int main(){
    cin >> n >> m;
    for(int i = 0; i < n; i ++ ){
        int x, c;   //在下标为x的地方加上c
        cin >> x >> c;
        
        add.push_back({x, c});
        v.push_back(x);
    }
    
    for(int i = 0; i < m; i ++ ){
        int l, r;   //存查询的左右区间
        cin >> l >> r;
        query.push_back({l, r});
        
        v.push_back(l);
        v.push_back(r);
        
    }
    
    // 对v数组排序、去重
    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()), v.end());
    
    // 处理插入
    for(auto item : add){
        int x = find(item.first);   //找到下标
        a[x] += item.second;
    }
    
    // 预处理前缀和
    for(int i = 1; i <= v.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] + a[l] << endl;
    }
   
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值