leet-code/src/main/java/leetcode/editor/cn/FindTheStudentThatWillReplaceTheChalk.java

//一个班级里有 n 个学生，编号为 0 到 n - 1 。每个学生会依次回答问题，编号为 0 的学生先回答，然后是编号为 1 的学生，以此类推，直到编号为 
//n - 1 的学生，然后老师会重复这个过程，重新从编号为 0 的学生开始回答问题。 
//
// 给你一个长度为 n 且下标从 0 开始的整数数组 chalk 和一个整数 k 。一开始粉笔盒里总共有 k 支粉笔。当编号为 i 的学生回答问题时，他会消耗
// chalk[i] 支粉笔。如果剩余粉笔数量 严格小于 chalk[i] ，那么学生 i 需要 补充 粉笔。 
//
// 请你返回需要 补充 粉笔的学生 编号 。 
//
// 
//
// 示例 1： 
//
// 输入：chalk = [5,1,5], k = 22
//输出：0
//解释：学生消耗粉笔情况如下：
//- 编号为 0 的学生使用 5 支粉笔，然后 k = 17 。
//- 编号为 1 的学生使用 1 支粉笔，然后 k = 16 。
//- 编号为 2 的学生使用 5 支粉笔，然后 k = 11 。
//- 编号为 0 的学生使用 5 支粉笔，然后 k = 6 。
//- 编号为 1 的学生使用 1 支粉笔，然后 k = 5 。
//- 编号为 2 的学生使用 5 支粉笔，然后 k = 0 。
//编号为 0 的学生没有足够的粉笔，所以他需要补充粉笔。 
//
// 示例 2： 
//
// 输入：chalk = [3,4,1,2], k = 25
//输出：1
//解释：学生消耗粉笔情况如下：
//- 编号为 0 的学生使用 3 支粉笔，然后 k = 22 。
//- 编号为 1 的学生使用 4 支粉笔，然后 k = 18 。
//- 编号为 2 的学生使用 1 支粉笔，然后 k = 17 。
//- 编号为 3 的学生使用 2 支粉笔，然后 k = 15 。
//- 编号为 0 的学生使用 3 支粉笔，然后 k = 12 。
//- 编号为 1 的学生使用 4 支粉笔，然后 k = 8 。
//- 编号为 2 的学生使用 1 支粉笔，然后 k = 7 。
//- 编号为 3 的学生使用 2 支粉笔，然后 k = 5 。
//- 编号为 0 的学生使用 3 支粉笔，然后 k = 2 。
//编号为 1 的学生没有足够的粉笔，所以他需要补充粉笔。
// 
//
// 
//
// 提示： 
//
// 
// chalk.length == n 
// 1 <= n <= 10⁵ 
// 1 <= chalk[i] <= 10⁵ 
// 1 <= k <= 10⁹ 
// 
// Related Topics 数组 二分查找 前缀和 模拟 👍 4 👎 0

package leetcode.editor.cn;

//1894:找到需要补充粉笔的学生编号
class FindTheStudentThatWillReplaceTheChalk {
    public static void main(String[] args) {
        //测试代码
        Solution solution = new FindTheStudentThatWillReplaceTheChalk().new Solution();
    }

    //力扣代码
    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        //    public int chalkReplacer(int[] chalk, int k) {
//        int temp = k;
//        int length = chalk.length;
//        int[] free = new int[length];
//        for (int i = 0; i < length; i++) {
//            if (i == 0) {
//                free[i] = chalk[i];
//            } else {
//                free[i] += chalk[i] + free[i - 1];
//            }
//            temp -= chalk[i];
//            if (temp < 0) {
//                return i;
//            }
//        }
//        temp %= free[length - 1];
//        if (temp == 0) {
//            return 0;
//        }
//        for (int i = 0; i < length; i++) {
//            temp -= chalk[i];
//            if (temp < 0) {
//                return i;
//            }
//        }
//        return 0;
//    }
        public int chalkReplacer(int[] chalk, int k) {
            int size = chalk.length;
            int sum = 0;
            int i = 0;
            for (; i < size && sum <= k; i++) {
                sum += chalk[i];
            }
            if (i < size) {
                return i - 1;
            }
            k %= sum;
            i = 0;
            for (; i < size && k >= 0; i++) {
                k -= chalk[i];
            }
            return i - 1;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}