找到需要补充粉笔的学生编号

/*
 * 思路：
 * 使用 Java Stream 来简化计算粉笔总数和寻找需要补充粉笔的学生。
 * 先计算总粉笔使用量，再用剩余的粉笔量逐个学生判断。
 */
import java.util.stream.IntStream;

public class Solution {
    public int chalkReplacer(int[] chalk, int k) {
        // 计算总粉笔使用量
        long total = IntStream.of(chalk).asLongStream().sum();
        // 计算剩余的粉笔量
        k %= total;
        // 模拟每个学生使用粉笔，找到需要补充粉笔的学生
        return IntStream.range(0, chalk.length)
                .filter(i -> {
                    if (k < chalk[i]) {
                        return true;
                    }
                    k -= chalk[i];
                    return false;
                })
                .findFirst()
                .orElse(-1); // 这个返回值理论上不会被执行
    }
}