无法吃午餐的学生数量

2020-12-27

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-students-unable-to-eat-lunch
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

因为学生的数量比较少，所以直接暴力模拟，不新建队列

class Solution {

    public int countStudents(int[] students, int[] sandwiches) {
        if (students.length == 0) return 0;
        if (sandwiches.length == 0 && students.length > 0) return students.length;
        boolean[] stu = new boolean[students.length];                               //指示是否已经吃到午餐
        int stuPoint = 0;                                                           //当前学生的指针
        int sanPoint = 0;                                                           //当前午餐的指针
        int stuLen = students.length;                                               //学生的数量
        int sanLen = sandwiches.length;                                             //午餐的数量

        int stuCount = stuLen;                                                      //当前没吃到午餐的学生的数量
        boolean stepFlag = true;
        int step = 0;                                                               //指示是否走了一圈但是没人能吃到午餐，如果大于stuLen就说明这一圈谁也没吃到
        while (stepFlag) {
            if (step > stuLen) {                                                    //走了一圈没人吃到，那么结束         
                return stuCount;
            }
            int stuIndex = stuPoint % stuLen;                                       //当前学生的index
            int sanIndex = sanPoint % sanLen;                                       //当前午餐的index
            while (stu[stuIndex]) {                                                 //如果已经吃到午餐了，就换下一个
                stuPoint++;
                stuIndex = stuPoint % stuLen;
            }
            if (!stu[stuIndex] && students[stuIndex] == sandwiches[sanIndex]) {     //如果这个午餐符合学生的胃口，就被吃掉

                stu[stuIndex] = true;
                stuPoint++;                                                         //学生和午餐指针同时步进1
                sanPoint++;
                step = 0;                                                           //走的数量归0
                stuCount--;                                                         //没吃到午餐的人的数量减1
                if (stuCount == 0) return 0;                                        //所有学生都吃到午餐了，那么结束
            } else {
                stuPoint++;
            }

            step++;
        }
        return 0;
    }
}

维护喜欢每种三明治的人的数量，当喜欢某个种类的三明治的人的数量为0时，所有的人都不喜欢栈顶部的三明治，剩下的人都吃不长了

class Solution {
    public int countStudents(int[] students, int[] sandwiches) {
        if (students == null || students.length == 0) return 0;
        if (sandwiches == null || sandwiches.length == 0) return students.length;
        // stuLike[0] 喜欢圆形的学生数，stuLike[1]喜欢方形的学生数
        int[] stuLike = new int[2];
        for (int student : students) {
            stuLike[student]++;
        }
        for (int i = 0; i < sandwiches.length; i++) {
            if (stuLike[sandwiches[i]] == 0) {
                // 学生是转圈来的，此时说明喜欢圆形或者方形的学生已经没有了，顶部的三明治没人取走了。
                break;
            }
            // 对应的学生满足一个，减少一个
            stuLike[sandwiches[i]]--;
        }
        return stuLike[0] + stuLike[1];
    }
}

参考文献

作者：bo-he-f 来源：力扣（LeetCode）著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。