当前位置：网站首页>LeetCode 1894.找到需要补充粉笔的学生编号

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

2022-08-10 05:32:00 【生信研究猿】

python

class Solution:
    def chalkReplacer(self, chalk: List[int], k: int) -> int:
        
        sum_chalk = 0
        for i in range(len(chalk)):
            sum_chalk += chalk[i]
        
        if(k>sum_chalk):
            k = k%sum_chalk
        
        i = 0
        while(1):
            k -= chalk[i%len(chalk)]
            # print(k)
            if(k<0):
                break
            i +=1
        return  i%len(chalk)