[OD E 100] 最大社交距离

题目

题目描述
疫情期间需要大家保证一定的社交距离，公司组织开交流会议。座位一排共 N 个座位，编号分别为[0,N-1],
要求员工一个接着一个进入会议室，并且可以在任何时候离开会议室。
满足：
每当一个员工进入时，需要坐到最大社交距离（最大化自己和其他人的距离的座位）；
如果有多个这样的座位，则坐到索引最小的那个座位。

输入描述
会议室座位总数 seatNum
1 ≤ seatNum ≤ 500
员工的进出顺序 seatOrLeave 数组
元素值为 1，表示进场
元素值为负数，表示出场（特殊：位置 0 的员工不会离开）
例如 -4 表示坐在位置 4 的员工离开（保证有员工坐在该座位上）

输出描述
最后进来员工，他会坐在第几个位置，如果位置已满，则输出-1。

示例1
输入

10
[1, 1, 1, 1, -4, 1]

输出

5

说明
seat -> 0,空在任何位置都行，但是要给他安排索引最小的位置，也就是座位 0
seat -> 9,要和旁边的人距离最远，也就是座位 9
seat -> 4,要和旁边的人距离最远，应该坐到中间，也就是座位 4
seat -> 2,员工最后坐在 2 号座位上
leave[4], 4 号座位的员工离开
seat -> 5,员工最后坐在 5 号座位上

思路

注意对这个社交距离的理解：和两侧相邻的人的距离的最小值
例如上面的例子中 0 4 9 这种情况，选择 2 而不是 6，虽然6左侧1个位置右侧2个位置

代码

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

using namespace std;

int findSeat(vector<int> &occupiedSeats, int seatNum)
{
	int n = occupiedSeats.size();
	int maxDistance = -1;
	int seat = -1;
	if (n == 0)
	{
		return 0;
	}
	if (n == 1)
	{
		return seatNum - 1;
	}

	if (occupiedSeats.front() > 0)
	{
		maxDistance = occupiedSeats.front() - 1;
		seat = 0;
	}
	if (occupiedSeats.back() < n - 1)
	{
		if (maxDistance < (seatNum - occupiedSeats.back() - 2))
		{
			maxDistance = seatNum - occupiedSeats.back() - 2;
			seat = seatNum - 1;
		}
	}

	for (int i = 0; i < n - 1; i++)
	{
		int distance = (occupiedSeats[i + 1] - occupiedSeats[i] - 1) / 2 - occupiedSeats[i] - 1;
		if (distance > maxDistance)
		{
			maxDistance = distance;
			seat = (occupiedSeats[i + 1] + occupiedSeats[i]) / 2;
		}
	}

	return seat;
}

int main()
{
	int n;
	cin >> n >> ws;
	string line;
	getline(cin, line);
	line = line.substr(1, line.size() - 2);
	cout << line << endl;
	stringstream ss(line);
	vector<int> vec;

	while (ss)
	{
		int item;
		ss >> item;
		vec.push_back(item);
		if (ss.peek() == ',')
		{
			ss.ignore();
		}
	}

	vector<int> occupiedSeats;
	int ans = -1;

	for (int i = 0; i < vec.size(); i++)
	{
		if (vec[i] == 1)
		{
			int seat = findSeat(occupiedSeats, n);
			if (seat != -1)
			{
				occupiedSeats.push_back(seat);
				sort(occupiedSeats.begin(), occupiedSeats.end());
				ans = seat;
			}
		}
		else if (vec[i] < 0)
		{
			for (auto beg = occupiedSeats.begin(); beg != occupiedSeats.end(); beg++)
			{
				if (*beg == abs(vec[i]))
				{
					occupiedSeats.erase(beg);
					break;
				}
			}
		}
	}

	cout << ans << endl;

	return 0;
}