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

题目描述

一个班级里有 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 的学生没有足够的粉笔，所以他需要补充粉笔。

解题思路

• 已知条件如下：

  • 粉笔总数 k；
  • 学生数组；
  • 每个学生所消耗的粉笔个数(学生下标对应的数组元素)；
  • 按学生编号消耗粉笔，老师会重复此过程

• 求：需要补充粉笔的学生编号（下标）

• 思考：

  • 已知总粉笔数量chalkCount; 计算出每一轮消耗的粉笔数量costCount， chalkCount%costCount
  • chalk = [5,1,5]
  • preSum = [5,6,11]
  • k=22
  • 22%11 = 0
  • 目标值为0，在数组preSum使用二分查找法寻找到的下标即为需要更换粉笔的学生编号

  总结

  解题的根本思路是：前缀和+二分查找

代码实现

public int getChalkSupplementIndex(int[] chalk, int k) {
        /**通过前缀和 求出每轮消耗的粉笔数量*/
        int n = chalk.length;
        int[] preSum = new int[n];
        preSum[0] = chalk[0];
        for (int i = 1; i < n; i++) {
            preSum[i] = preSum[i-1] + chalk[i];
        }
        //取模，作为目标值，
        k %= preSum[n-1];
        /**通过二分查找法在前缀和数组中寻找下标，即为需要补充的学生编号*/
        int left = 0, right = preSum.length-1;
        while (left < right){
            int mid = (left + right) >> 1;
            if(preSum[mid] > k){
                right = mid ;
            }else {
                left = mid + 1;
            }
        }
        return left;
    }