Awcing算法---区间和(离散化)

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int ,int> pii; //定义 pair<int,int> 名字为 pii
const int N=300030;  //alls总共要存储出现的所有下标,把它们全部离散化映射。 x,l,r 总共3*100000
int a[N],s[N]; //a[]数组是计算值  s[]数组是前缀和数组
vector<int> alls; //alls记录所有下标
vector<pii> num,add;//add记录<x,c>     num记录左右端点<l,r>
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 l+1;//之所以+1,是为了 使a[0]=0,s[0]=0. 前缀和算法
}
int main(){
    int n,m;
    cin >>n>>m;
    int x,c;
    for(int i=0;i<n;i++){ //读入
        cin >>x>>c;
        add.push_back({x,c});
        alls.push_back(x);
    }
    int l,r;
    //读入每次的 左右端点
    for(int i=0;i<m;i++){
        cin >>l>>r;
        num.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){
        a[find(item.first)]+=item.second;
    }
    //求前缀和
    for(int i=1;i<=alls.size();i++){
        s[i]=s[i-1]+a[i];
    }
    //求l--r的和,使用前缀和算法
    for(auto item:num){
        l=find(item.first);
        r=find(item.second);
        cout <<s[r]-s[l-1]<<endl;
    }
    return 0;
}

//unique()函数实现
// vector<int> ::iterator unique(vector<int> &A){
//     int i,j=0;
//     for(i=0;i<A.size();i++){
//         if(!i||A[i]!=A[i-1]){
//             A[j++]=A[i];
//         }
//     }
//     return j+A.begin();
// }

1.读输入。将每次读入的x c push_back()到add中,将每次读入的位置x push_back()到alls中,将每次读入的l r push_back()到query中。
2.排序、去重。
3.通过遍历add,完成在离散化的数组映射到的a数组中进行加上c的操作(用到find函数)。
4.初始化s数组。
5.通过遍历query,完成求区间[l,r]的和。

vector<int> alls; // 存储所有待离散化的值
sort(alls.begin(), alls.end()); // 将所有值排序
alls.erase(unique(alls.begin(), alls.end()), alls.end());   // 去掉重复元素

// 二分求出x对应的离散化的值
int find(int x) // 找到第一个大于等于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, 2, ...n
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值