持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第11天，点击查看活动详情

题目

学校的自助午餐提供圆形和方形的三明治，分别用数字 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 。

题解

我们观察发现，学生位置可调整，而三明治位置不可调整。也就是说，若前面的三明治没被拿走，则往后的所有三明治也无法被拿走。

学生队列是循环的，不需要关心学生排队的顺序；只关注学生队列中 0 和 1 的数量即可。基于以上，计数两种三明治对应的学生数量，之后减少相应三明治数量，直到某种三明治的个数为 0 时，说明不再有学生会消耗三明治，返回此时剩余学生数量。

/**
 * @param {number[]} students
 * @param {number[]} sandwiches
 * @return {number}
 */
var countStudents = function(students, sandwiches) {
    let s1 = _.sum(students);
    let s0 = students.length - s1;
    for (let i = 0; i < sandwiches.length; i++) {
        if (sandwiches[i] === 0 && s0 > 0) {
            s0--;
        } else if (sandwiches[i] === 1 && s1 > 0) {
            s1--;
        } else {
            break;
        }
    }
    return s0 + s1;
};

代码详解

因此，我们先用计数器 cnt 统计学生喜欢的三明治种类和对应的数量。

然后遍历三明治，若在 cnt 中找不到喜欢此三明治的学生，说明当前及往后的三明治均无法被拿走，返回当前剩余的学生数量即可。

遍历结束，返回 0。

算法--无法吃午餐的学生数量

题目

题解

代码详解