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

题目描述

博艾市有一个木材仓库，里面可以存储各种长度的木材，但是保证没有两个木材的长度是相同的。作为仓库负责人，你有时候会进货，有时候会出货，因此需要维护这个库存。有不超过 100000 条的操作：

• 进货，格式1 Length：在仓库中放入一根长度为 Length(不超过 $10^9$) 的木材。如果已经有相同长度的木材那么输出Already Exist。
• 出货，格式2 Length：从仓库中取出长度为 Length 的木材。如果没有刚好长度的木材，取出仓库中存在的和要求长度最接近的木材。如果有多根木材符合要求，取出比较短的一根。输出取出的木材长度。如果仓库是空的，输出Empty。

输入格式

输出格式

样例 #1

样例输入 #1

7
1 1
1 5
1 3
2 3
2 3
2 3
2 3

样例输出 #1

3
1
5
Empty

算法1

(Set)

看完题目就想到了set

没想到的地方：

出货时，如何找更优解

主要思路

1.set有去重，排序的功能

2.进货时，set集合中是否已经存在该长度的木头，如果有则输出Already Exist，如果没有则插入该长度

if(op == 1){
    if(s.count(x))   //可以利用count(x)，来查找存在情况
    cout << "Already Exist" <<endl;
    else s.insert(x);
}

3.出货时，首先判断集合是否为空，如果不空那么通过s.lower_bound(x) ，来查找第一个大于等于x的值，注意返回值为迭代器

分两种情况:
(1)存在长度为x的木头时，直接输出

(2)不存在长度为x的木头

	if(s.empty()) cout << "Empty" << endl;
			else{
				iter l,r;
				r = s.lower_bound(x);  //返回第一个大于等于x的数的位置
				l = r;
				if(r!=s.begin()) l--;   //集合中存在一个元素以上的时候 
			    if(r!=s.begin() && abs(*l - x) <= abs(*r -x)) r--; 
				cout <<  *r << endl;
				s.erase(*r); 
			}  

C++ 代码

#include <bits/stdc++.h>
using namespace std;

typedef set<int>::iterator iter;

int n;
set<int>s;

int main(){
	cin >> n;
	while(n--){
		int op,x;
		cin >> op >> x;
		
		//插入操作 
		if(op == 1) {
		   if(s.count(x)) cout << "Already Exist" << endl;
		   else s.insert(x);
		}
		else{
			if(s.empty()) cout << "Empty" << endl;
			else{
				iter l,r;
				r = s.lower_bound(x);  //返回第一个大于等于x的数的位置
				l = r;
				if(r!=s.begin()) l--;   //集合中存在一个元素以上的时候 
			    if(r!=s.begin() && abs(*l - x) <= abs(*r -x)) r--; 
				cout <<  *r << endl;
				s.erase(*r); 
			} 
		}
	}
	return 0;
}