无法吃午餐的学生数量

/*
 * 思路: 
 * 1. 使用 Java Stream API 统计学生喜欢的三明治的类型数量。
 * 2. 依次查看三明治栈顶的三明治类型，若有学生喜欢该类型，则移除对应数量。
 * 3. 若无学生喜欢，直接返回剩余的学生数量。
 */
import java.util.Arrays;

public class Solution {
    public int countStudents(int[] students, int[] sandwiches) {
        long count0 = Arrays.stream(students).filter(s -> s == 0).count();
        long count1 = students.length - count0;
        for (int sandwich : sandwiches) {
            if (sandwich == 0) {
                if (count0 == 0) return (int) count1;
                count0--;
            } else {
                if (count1 == 0) return (int) count0;
                count1--;
            }
        }
        return 0;
    }
}