1894.找到需要补充粉笔的学生编号
题目:
来自LeetCode
一个班级里有 n 个学生,编号为 0 到 n - 1 。每个学生会依次回答问题,编号为 0 的学生先回答,然后是编号为 1 的学生,以此类推,直到编号为 n - 1 的学生,然后老师会重复这个过程,重新从编号为 0 的学生开始回答问题。
给你一个长度为 n 且下标从 0 开始的整数数组 chalk 和一个整数 k 。一开始粉笔盒里总共有 k 支粉笔。当编号为 i 的学生回答问题时,他会消耗 chalk[i] 支粉笔。如果剩余粉笔数量 严格小于 chalk[i] ,那么学生 i 需要 补充 粉笔。
请你返回需要 补充 粉笔的学生 编号 。
示例:
输入: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 的学生没有足够的粉笔,所以他需要补充粉笔。
解题思路:
一开始用暴力算法,用双循环直接拿粉笔总数一个一个减去chalk数组每一个数,直到chalk[i]要大于总数k。
后来发现时间复杂度太高,过不去。
如果时间复杂度太高,我们可以想办法拆分循环
如:o(n二次方) 可以变成o(n+m) o(n三次方) 可以变成o(n二次方+n)
我们将所给列表chalk加起来,
如果k/(chalk所有数相加)小于1,我们就遍历一次chalk,k -=chalk[i],直到k小于chalk[i]
如果k/(chalk所有数相加)大于1,我们就让k%(chalk所有数加和),取得余数remain_chalk后,遍历一次chalk,remain_chalk -= chalk[i],直到remain_chalk小于chalk[i]
代码:
target = 0
need_chalk = 0
for i in range(len(chalk)):
need_chalk += chalk[i] # 获得一组chalk所需要的粉笔数
turn_num = k / need_chalk # 获得轮数
if turn_num < 1: # 轮数小于1,直接遍历chalk直到取到i
for i in range(len(chalk)):
if k>=chalk[i]:
k -= chalk[i]
elif k<chalk[i]:
target = i
return target
elif turn_num == 1:
return 0 # 轮数等于1,刚好在第二组chalk[0]没有粉笔用
else:
# 轮数大于1,对k取need_chalk的余为remain_chalk,遍历chalk,直到remain_chalk < chalk[j]
remain_chalk = k % need_chalk
for j in range(len(chalk)):
if remain_chalk >= chalk[j]:
remain_chalk -= chalk[j]
else:
target = j
return target
优化算法的目的就是为了让时间空间复杂度变得更小,遇到能拆分复杂循环的可以尽量拆分,这对提高编程能力有很大帮助。