802. 区间和

题目:

代码:
 

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

const int N = 300010;

int a[N],s[N];//a数组用来存映射后的元素,这些元素原先的坐标包括x,l,r
typedef pair<int ,int > PII;//pair<int,int>类似于结构体的简写版，typedef定义了一个新的类型PII(跟结构体定义了一个结构体类型然后使用相似)
//将不按照原先排列顺序的旧坐标(此时旧坐标是alls中的元素)按照大小排好序后元素对应的新位置就是原坐标投射后的新坐标
//投射后的原坐标改变了绝对位置,但是相对位置没有改变,所以求新区间的前缀和不会对答案有影响
vector<int> alls;
vector<PII> add,query;
//二分查找 用来找出x在已排好序的alls数组中的新位置
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 r+1;//因为后续要使用前缀和，所以返回的坐标要加上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);
    }


//将alls排好序,并且删除重复操作(如进行了两次在x的增值操作，应该去掉一个x保持平衡)
sort(alls.begin(),alls.end());
 alls.erase(unique(alls.begin(),alls.end()),alls.end());
 
  //一个迭代器从1开始直到末尾结束，itdm.first是x，second是c
  //进行插入操作
 for(auto itdm :add){
     int x;
     x=find(itdm.first);//找到x在alls中的新坐标
     a[x] += itdm.second;
 }

  for(int i=1;i<=alls.size();i++) s[i]=s[i-1]+a[i];//求出a数组的前缀和,a数组相当于把alls数组插入若干个操作后复制给a

    for(auto itdm : query)
    {
        int l,r;
        l=find(itdm.first);//找出l，r在alls或者a中的坐标
        r=find(itdm.second);
        printf("%d\n",s[r]-s[l-1]);//前缀和上方已计算，所以可直接输出，记得加上换行符！
    }

    return 0;
}