华为od（D卷）最大社交距离

题目描述

疫情期间需要大家保证一定的社交距离，公司组织开交流会议。

座位一排共 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，不是7；
所有社交距离的理解应该是：距离相近左右两人的距离最小值;也就是说 位置2的社交距离是2和位置6，位置7的社交距离是相同的，所以才选择了索引小的2。

那代码上面如何体现呢？ 相邻座位的距离如果相等，或者相差不超过1，那么中间位置的最大社交距离应该是相同的。

代码

public class Demo05 {
    public static void main(String[] args) {
        // 座位数
        int seatNum = 10;
        // 坐下或者离开
        int[] seatOrLeave = new int[] {1, 1, 1, 1, -4, 1};
        //做下的位子编号
        TreeSet<Integer> seated = new TreeSet<>();
        for (int i = 0; i < seatOrLeave.length; i++) {
            int sOrL = seatOrLeave[i];
            if (sOrL > 0) {
                // 坐下的位置
                int seatIndex = -1;
                if (seated.size() == 0) {
                    seatIndex = 0;
                } else if (seated.size() == 1) {
                    seatIndex = seatNum - 1;
                } else {
                    seatIndex = center(seated);
                }

                if (seatIndex == -1) {
                    System.out.println(seatIndex);
                    return;
                } else {
                    seated.add(seatIndex);
                }

                //最后一个人
                if (i == seatOrLeave.length - 1) {
                    System.out.println(seatIndex);
                }

            } else {
                // 离开
                seated.remove(i);
            }
        }

    }

    // 查找中间位置
    public static int center(TreeSet<Integer> seated) {
        // 两个位置之间的距离最大值
        int maxLength = 0;
        // 两个位置之间的距离最大值时，左侧坐下人位置
        int index = -1;
        // 遍历时前一个人坐下的位置
        int before = -1;
        for (Integer post : seated) {
            if (before == -1) {
                before = post;
                continue;
            }

            Integer l = post - before;
            // 中间没座位
            if (l == 1) {
                before = post;
                continue;
            }
            // 相邻座位的距离如果相等，或者相差不超过1，那么中间位置的最大社交距离应该是相同的。
            if (l > maxLength + 1) {
                maxLength = l;
                index = before + maxLength / 2;
            }
            before = post;

        }
        return index;
    }
}