Cộng đồng Hỏi & Đáp của Inflearn

rocks = [1, 2, 1, 4]
rabbits = [2, 1, 3]

def sol(rocks, rabbits):
  results = ["pass" for i in range(len(rabbits))]
  
  for i in range((len(rabbits))):
    step = 0

    while step < len(rocks):
      step += rabbits[i]
      print(f'step: {step}')

      if step <= len(rocks):
        rocks[step - 1] -= 1

        if rocks[step - 1] < 0:
          results[i] = 'fail'

      print(rocks)
    print('---------------')

  return results

sol(rocks, rabbits)

배열 크기를 벗어나면 오류가 생겨서 if문으로 걸러줬습니다.