基础算法6:离散化

问题引入： a[i]:1 3 100 2000 50000 映射到 0 1 2 3 4

解决问题1：a[]中可能重复元素

解决问题2：如何快速算出x离散化后的值

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<vector>

/*
0 1 2 3 4 5 6 7 8 9
0 2 0 6 0 0 0 5 0 0
 (1,3): 2+0+6=8
 (4,6): 0+0+0=0
 (7,8): 5+0=5
 */
using namespace std;

const int N=300010;


typedef pair<int,int> PII;

//n个添加操作，m个询问操作
int n,m;

//a数组存数，S数组存前缀和
int a[N],S[N];

//所有会用到的元素下标
vector<int> alls;
//添加操作，查询操作
vector<PII> add,query;

//找到第一个大于等于x的位置
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
}
int main()
{
    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);
    }
    sort(alls.begin(),alls.end());
    alls.erase(unique(alls.begin(),alls.end()),alls.end());
    for(auto item:add)
    {
        int x=find(item.first);
        a[x]+=item.second;
    }
    for(int i=1;i<=alls.size();i++) S[i]=S[i-1]+a[i];
    for(auto item:query)
    {
        int l=find(item.first),r=find(item.second);
        cout<<S[r]-S[l-1]<<endl;
    }

    return 0;
}