洛谷 P5250 【深基17.例5】木材仓库(集合：查找+删除+插入(vector写法))(二分查找)

P5250 【深基17.例5】木材仓库

菜鸟生成记(50)

这题数据不是很大vector就能水过,map理论上应该是更快一些;

但是不太好写(map理解了，但是还没完全理解0_0);

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<set>
using namespace std;
int find_lower(int st,int ed,int num,vector<int> &a)
{//查找比num小的最后一个数 
	if(st>ed)
	{
		if(st==0)
		return -1;
		else
		return st-1;//返回下标 
	}
	int mid=(st+ed)/2;
	if(a[mid]<=num)//右边查找 
	find_lower(mid+1,ed,num,a);
	else if(a[mid]>num)//左边查找 
	find_lower(st,mid-1,num,a);
}
int find_upper(int st,int ed,int num,vector<int> &a)
{//查找比num大的第一个数;
	if(st>ed)
	{
		if(ed==a.size()-1)
		return -1;
		else
		return ed+1;//返回下标
	}
	int mid=(st+ed)/2;
	if(a[mid]<num)//右边查找
	find_upper(mid+1,ed,num,a);
	else if(a[mid]>=num)//左边查找 
	find_upper(st,mid-1,num,a);
}
int main()
{
	int n;
	cin>>n;
	int lower,upper;
	vector<int>a;
	vector<int>::iterator it,j,k,s;
	for(int i=0;i<n;i++)
	{
		int x,num;
		cin>>x>>num;
		switch(x)
		{
			case 1://插入 num 
				it=find(a.begin(),a.end(),num);
				if(it!=a.end())//num已存在 
				{
					cout<<"Already Exist"<<endl;
				}
				else//num不存在 
				{//lower_bound():库函数(查找大于或等于num的第一个数) 
				//然后插在其前面,在这样的规则下插入的数据一直保持升序
				//方便后面的二分查找 
					a.insert(lower_bound(a.begin(),a.end(),num),num);
				}
				break;
			case 2://取出 num 
				it=find(a.begin(),a.end(),num);//在容器a中查找num 
				if(a.empty())//容器空 
				{
					cout<<"Empty"<<endl;
				}
				else//不空 
				{
					if(it!=a.end())//找到num，取出num，并在容器中删除num
					{
						cout<<*it<<endl;
						a.erase(it);
					}
					else if(it==a.end())//未找到num，找num的前后（前后与num差值小的） 
					{
					
					//库函数:upper_bound()和lower_bound(),只能查找大于或等于查找值的数 
				 	//所以手写了两个个二分查找 ,查找效率还可以(O(lgn)) 
				 		lower=find_lower(0,a.size()-1,num,a);//查找最后一个小于num的数的下标;
				 		upper=find_upper(0,a.size()-1,num,a);//查找第一个大于num的数的下标;
				 		if(upper==-1)
				 		{//没有比num大的数了
						 //输出最后一个小于num的数;
				 			cout<<a[lower]<<endl;
				 			a.erase(a.begin()+lower);//输出后删除 
						}
				 		else if(lower==-1)
				 		{//没有比num小的数了 
				 		//第一个大于num的数;
				 			cout<<a[upper]<<endl;
							 a.erase(a.begin()+upper);//输出后删除 	
						}
						else//两边的数都查找到了 
						{//然后比较谁离num更近 
							int le=(num-a[lower])<=(a[upper]-num)?lower:upper;
							cout<<a[le]<<endl;//输出 
							a.erase(a.begin()+le);输出后删除 	
						}
					}
				}
			break; 
		}
	}
	return 0;
 }