简单

问题描述

学校的自助午餐提供圆形和方形的三明治，分别用数字 0 和 1 表示。所有学生站在一个队列里，每个学生要么喜欢圆形的要么喜欢方形的。 餐厅里三明治的数量与学生的数量相同。所有三明治都放在一个 栈 里，每一轮：

• 如果队列最前面的学生 喜欢 栈顶的三明治，那么会 拿走它 并离开队列。
• 否则，这名学生会 放弃这个三明治 并回到队列的尾部。

这个过程会一直持续到队列里所有学生都不喜欢栈顶的三明治为止。

给你两个整数数组 students 和 sandwiches，其中 sandwiches[i] 是栈里面第 i 个三明治的类型（i = 0 是栈的顶部），students[j] 是初始队列里第 j 名学生对三明治的喜好（j = 0 是队列的最开始位置）。请你返回无法吃午餐的学生数量。

示例 1：

输入：students = [1,1,0,0], sandwiches = [0,1,0,1]
输出：0
解释：
- 最前面的学生放弃最顶上的三明治，并回到队列的末尾，学生队列变为 students = [1,0,0,1]。
- 最前面的学生放弃最顶上的三明治，并回到队列的末尾，学生队列变为 students = [0,0,1,1]。
- 最前面的学生拿走最顶上的三明治，剩余学生队列为 students = [0,1,1]，三明治栈为 sandwiches = [1,0,1]。
- 最前面的学生放弃最顶上的三明治，并回到队列的末尾，学生队列变为 students = [1,1,0]。
- 最前面的学生拿走最顶上的三明治，剩余学生队列为 students = [1,0]，三明治栈为 sandwiches = [0,1]。
- 最前面的学生放弃最顶上的三明治，并回到队列的末尾，学生队列变为 students = [0,1]。
- 最前面的学生拿走最顶上的三明治，剩余学生队列为 students = [1]，三明治栈为 sandwiches = [1]。
- 最前面的学生拿走最顶上的三明治，剩余学生队列为 students = []，三明治栈为 sandwiches = []。
  所以所有学生都有三明治吃。

示例 2：

输入：students = [1,1,1,0,0,1], sandwiches = [1,0,0,0,1,1]
输出：3

提示：

• 1 <= students.length, sandwiches.length <= 100
• students.length == sandwiches.length
• sandwiches[i] 要么是 0，要么是 1。
• students[i] 要么是 0，要么是 1。

模拟

实际上，三明治是有序的，但人可以任意顺序，因为必然能够通过任意次入队出队使得队首的人为想要的人，因此首先对人进行分组统计，统计出两组人的人数，然后遍历三明治数组，若当前三明治为 $1$，则分组 $1$ 的人数减一，若当前三明治为 $0$，则分组 $0$ 的人数减一，代表该学生吃到三明治，当某一块三明治无法分配到人时，即人数为 $0$，此时这个三明治永远不会被吃掉，那么此时剩下的人吃不上午餐。

/**
 * @param {number[]} students
 * @param {number[]} sandwiches
 * @return {number}
 */
var countStudents = function (students, sandwiches) {
  let O = 0
  let I = 0
  for (const student of students) {
    if (student === 0) O++
    else I++
  }
  for (const sandwich of sandwiches) {
    if (sandwich === 0 && O > 0) O--
    else if (sandwich === 1 && I > 0) I--
    else return O + I
  }
  return 0
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$