安排电影院座位Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
如上图所示,电影院的观影厅中有 n 行座位,行编号从 1 到 n ,且每一行内总共有 10 个座位,列编号从 1 到 10 。
给你数组 reservedSeats ,包含所有已经被预约了的座位。比如说,reservedSeats[i]=[3,8] ,它表示第 3 行第 8 个座位被预约了。
请你返回 最多能安排多少个 4 人家庭 。4 人家庭要占据 同一行内连续 的 4 个座位。隔着过道的座位(比方说 [3,3] 和 [3,4])不是连续的座位,但是如果你可以将 4 人家庭拆成过道两边各坐 2 人,这样子是允许的。
示例 1:
输入:n = 3, reservedSeats = [[1,2],[1,3],[1,8],[2,6],[3,1],[3,10]] 输出:4 解释:上图所示是最优的安排方案,总共可以安排 4 个家庭。蓝色的叉表示被预约的座位,橙色的连续座位表示一个 4 人家庭。 示例 2:
输入:n = 2, reservedSeats = [[2,1],[1,8],[2,6]]
输出:2
示例 3:
输入:n = 4, reservedSeats = [[4,3],[1,4],[4,6],[1,7]]
输出:4
提示:
- 1 <= n <= 10^9
- 1 <= reservedSeats.length <= min(10*n, 10^4)
- reservedSeats[i].length == 2
- 1 <= reservedSeats[i][0] <= n
- 1 <= reservedSeats[i][1] <= 10
- 所有 reservedSeats[i] 都是互不相同的。
# 思路
HashMap<Integer, Set<Integer>> map = new HashMap<>();
# 解法
class Solution {
public int maxNumberOfFamilies(int n, int[][] reservedSeats) {
HashMap<Integer, Set<Integer>> map = new HashMap<>();
for (int[] reservedSeat : reservedSeats) {
Set<Integer> set = map.getOrDefault(reservedSeat[0], new HashSet<>());
set.add(reservedSeat[1]);
map.put(reservedSeat[0], set);
}
int ans = 2 * n;
for (Integer row : map.keySet()) {
Set<Integer> set = map.get(row);
int cnt = 0;
//考虑2,5之间知否有
if (isOk(2, 5, set)) {
cnt++;
}
//考虑6,9之间是否有
if (isOk(6, 9, set)) {
cnt++;
//这里还有一种特殊情况:就是2,5有,但是都在最左边;6,9有,但是都在最右边,而此时4,7就符合条件,不需要减去
if (isOk(2, 5, set) && !isOk(4, 7, set)) cnt--;
}
ans -= cnt;
}
return ans;
}
private boolean isOk(int i, int i1, Set<Integer> set) {
for (int j = i; j <= i1; j++) {
if (set.contains(j)) return true;
}
return false;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# 总结
- 分析出几种情况,然后分别对各个情况实现