leetcode1894. 找到需要补充粉笔的学生编号 -增量数组-取余-二分查找-CSDN博客

本文链接：https://blog.csdn.net/Julse/article/details/120217497

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-student-that-will-replace-the-chalk

解法1 - 穷举- 超时

'''
超时
'''
class Solution:
    def chalkReplacer(self, chalk, k) -> int:
        i = -1
        while k>=0:
            i+=1
            i = i%len(chalk)
            k-=chalk[i]
        return i

解法2 - 增量数组-取余 - 通过

数组的和小于粉笔总数，通过取余的方式来索引添粉笔的学生数目

'''
pass
'''
class Solution:
    def chalkReplacer(self, chalk, k) -> int:
        i = 0
        for i in range(1,len(chalk)):
            if chalk[i-1]>k:return i-1
            chalk[i]+=chalk[i-1]
        if chalk[i-1] > k: return i - 1
        ret = k%chalk[len(chalk)-1]
        for i in range(len(chalk)):
            if chalk[i]>ret:return i
        return 0

解法3- 增量数组-取余-二分查找

'''
pass
'''
class Solution:
    def chalkReplacer(self, chalk, k) -> int:
        i = 0
        for i in range(1,len(chalk)):
            if chalk[i-1]>k:return i-1
            chalk[i]+=chalk[i-1]
        if chalk[i-1] > k: return i - 1
        ret = k%chalk[len(chalk)-1]
        if ret ==0:return 0
        i = 0
        j = len(chalk)-1
        mid = 0
        while(i<=j):
            mid = (i+j)//2
            if chalk[mid]<ret:
                i = mid+1
            if chalk[mid]>ret:
                j = mid-1
            if chalk[mid]==ret:return mid+1
        return mid+1 if chalk[mid]<ret else mid