离散化 AcWing 802. 区间和

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

using namespace std;

typedef pair<int,int> PII;//定义pir为 两个int数组 
const int N=3e5+10;//l,r,x每个是1e5 
int n,m,a[N],s[N];
vector<int> alls;
vector <PII> add,query;

int find (int x)
{
    
    
	int l=0,r=alls.size()-1;
	while(l<r)
	{
    
    
		int mid=r+l>>1;
		if(alls[mid]>=x) r=mid;
		else l=mid+1;
	}
	return r+1;//对应下面53行 
}

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());//去掉重复数组,unique返回的是重复的第一个值,erase是擦除 
	
	for(int i=0;i<add.size();i++)
	{
    
    
		PII item=add[i];
		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(int i=0;i<query.size();i++)
	{
    
    
		PII item=query[i];	
		int l=find(item.first),r=find(item.second);
		cout<<s[r]-s[l-1]<<endl;
	}
 	return 0;	
}
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
}

猜你喜欢

转载自blog.csdn.net/qq_47874905/article/details/109732097