HNU程序设计训练2023——占座位

【问题描述】sun所在学校的教室座位每天都是可以预占的。
一个人可以去占多个座位，而且一定是要连续的座位，如果占不到他所要求的这么多座位，那么他就一个座位也不要了。为了降低难度，每次分配座位按座位号从小到大查找，采用最先适配法分配座位。

【输入形式】输入有多组数据。
每组数据输入座位排数n，0<n<=100（座位的排列数相等，座位是按每行从左到右依次排序的, 第1行的最右边一个座位与第二行的第一个座位视为连续座位），m（ 0<m<=min(100,n*n) ）个人。
然后输入k（0<k<=100），最后输入k个命令。
命令只有两种：
1.in id num（代表id,0<=id<m,要占num个座位，若占不到连续的num(0<num<=20)个座位表示该命令无效）
2.out id（代表id要释放他之前占的所有座位）
注意：如果id之前占过座还没释放那么之后他的in命令都是无效的，
如果id之前没占过座位那么他的out命令也是无效的。

【输出形式】对每个in命令输出yes或者no，如果命令有效则输出yes，无效则输出no。
在yes no后面只带有回车，不带其他任何字符。

【样例输入】

4 10

9

in 1 7

in 2 3

in 3 3

in 3 3

in 4 3

out 2

in 5 6

out 3

in 5 6

【样例输出】

yes
yes

yes

no

yes

yes

no

yes

yes

#include <iostream>
#include <string>
#include <vector>
#include <map>
using namespace std;

int n = 0, m = 0;
void IN(vector<int>& seat, map<int, pair<int,int>>& stu)
{
	int id = 0, num = 0;
	cin >> id >> num;
	map<int, pair<int, int>>::iterator it;
	it = stu.find(id);
	if (it != stu.end() && it->second.second != 0)//该用户已经占座
	{
		cout << "no" << endl;
		return;
	}
	else
	{
		stu[id].second = 0;//初始化为0，占座成功后改为num
		for (int i = 0; i <= n * n - num; i++)
		{
			if (seat[i] == 0)
			{
				bool flag = true;
				for (int j = 0; j < num; j++)
				{
					if (seat[i+j] != 0)//无法满足连续的num个座位
					{
						flag = false;
						break;
					}
				}
				if (flag)//可以占座成功
				{
					seat[i] = id;//id占的起始位
					stu[id].first = i;//id占的座位的起始位置
					stu[id].second = num;
					cout << "yes" << endl;
					return;
				}
				else
					continue;//继续向下查找
			}
			else
				i += stu[seat[i]].second - 1;//下次循环前有i++，所以这里先-1
		}
		cout << "no" << endl;//在for循环中未找到可行座位
	}
}
void OUT(vector<int>& seat, map<int, pair<int, int>>& stu)
{
	int id = 0;
	cin >> id;
	map<int, pair<int, int>>::iterator it;
	it = stu.find(id);
	if (it == stu.end()|| stu[id].second == 0)//stu中没有id，他之前没有占过座,或者他目前占座数为0
	{
		cout << "no" << endl;
	}
	else
	{
		seat[stu[id].first] = 0;//删去seat中id的占座标识
		stu[id].second = 0;//id的占座数改为0
		cout << "yes" << endl;
	}
}
int main()
{
	while (cin >> n >> m)
	{
		vector<int> seat(n * n,0);//储存占座人的id
		map<int, pair<int,int>> stu;//id占座起始位置与数量
		int k = 0;
		cin >> k;
		string order;
		while (k--)
		{
			cin >> order;
			if (order == "in")
				IN(seat,stu);
			else
				OUT(seat, stu);
		}
	}
	return 0;
}