1894.找到需要补充粉笔的学生编号
难度:中等
题目
一个班级里有 n 个学生,编号为 0 到 n - 1 。每个学生会依次回答问题,编号为 0 的学生先回答,然后是编号为 1 的学生,
以此类推,直到编号为 n - 1 的学生,然后老师会重复这个过程,重新从编号为 0 的学生开始回答问题。
给你一个长度为 n 且下标从 0 开始的整数数组 chalk 和一个整数 k 。一开始粉笔盒里总共有 k 支粉笔。
当编号为 i 的学生回答问题时,他会消耗 chalk[i] 支粉笔。如果剩余粉笔数量 严格小于 chalk[i] ,那么学生 i 需要 补充 粉笔。
请你返回需要 补充 粉笔的学生 编号 。
提示:
- chalk.length == n
- 1 <= n <= 10 ^ 5
- 1 <= chalk[i] <= 10 ^ 5
- 1 <= k <= 10 ^ 9
示例
示例 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的取值范围是10 ^ 9所以,需要将K先对sum的chalk取余。
为什么?如果k = 10 ^ 9,而chalk == 1,这样循环起来时间就爆了...
来看看解题吧:
解题
Python:
class Solution: def chalkReplacer(self, chalk, k): k %= sum(chalk) if k == 0: return 0 for i, num in enumerate(chalk): k -= num if k < 0: return i
Java:
class Solution { public int chalkReplacer(int[] chalk, int k) { long sum = 0; int ret = 0; for (int i : chalk) sum += i; k %= sum; for (int i = 0; i < chalk.length; i++) { k -= chalk[i]; if (k < 0) return i; } return 0; } }