【LeetCode】Day162-无法吃午餐的学生数量

原创已于 2023-01-29 21:21:24 修改 · 171 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

于 2022-10-19 12:13:52 首次发布

Leetcode进阶之路专栏收录该内容

232 篇文章

订阅专栏

该博客介绍了如何通过简单的数量关系来解决一道关于学生和三明治匹配的问题。题目中，学生分为喜欢吃0号三明治和1号三明治两类，目标是计算在给定的三明治序列下，有多少学生无法吃到喜欢的三明治。博主尝试使用双端队列但未成功，最终发现只需维护两个计数器s0和s1，分别代表喜欢0号和1号三明治的学生，并遍历三明治序列进行匹配。时间复杂度和空间复杂度均为O(m+n)，其中m和n分别为学生和三明治的数量。

题目

1700. 无法吃午餐的学生数量【简单】

题解

看着题目是道双端队列，也试图用双端队列做，结果失败了

官解只用了数量关系判断就解决了…

假设吃“0”三明治的学生数量为s0，吃“1”三明治的学生数量为s1，我们只需要记录队列剩余学生中s0和s1的值。

如果栈顶为0，且s0>0，s0–；
如果栈顶为1，且s1>0，s1–；
否则终止过程，返回s0+s1。

class Solution {
    public int countStudents(int[] students, int[] sandwiches) {
        int m=students.length,n=sandwiches.length;
        int s0=0,s1=0;
        for(int i=0;i<m;i++){
            int like=students[i];
            if(like==0) s0++;
            else    s1++;
        }
        for(int i=0;i<n;i++){
            int like=sandwiches[i];
            if(s0>0&&like==0) s0--;
            else if(s1>0&&like==1)    s1--;
            else    return s0+s1;
        }
        return 0;
    }
}