Проблема с литкодом: https://leetcode.com/problems/eliminate-maximum-number-of-monsters/
Мое решение Python ниже:
class Solution(object):
def eliminateMaximum(self, dist, speed):
"""
:type dist: List[int]
:type speed: List[int]
:rtype: int
"""
mapped = []
killed = 0
for i, v in enumerate(zip(dist, speed)):
mapped.append([v[0]/v[1], v[0], v[1]])
mapped.sort(key=lambda x: x[0])
iterator = iter(mapped)
while True:
try:
next(iterator)[1] = -1
killed += 1
except StopIteration:
return killed
for v in mapped[killed:]:
v[1] = v[1]-v[2]
if v[1] <= 0:
return killed
sol = Solution()
r = sol.eliminateMaximum([4,8,6,8,2,7,4], [1,3,3,1,10,1,1])
print(r)
Вывод консоли Python равен 4 (как и ожидалось), а вывод Leetcode — 2, что является неправильным ответом.
🤔 А знаете ли вы, что...
Python активно используется в научных и инженерных вычислениях.
Python использует python2, в котором целочисленное деление отличается от python3. Вы можете увидеть информацию о языке здесь:
При выборе python3 это работает.
Я запустил этот код:
class Solution:
def eliminateMaximum(self, dist: List[int], speed: List[int]) -> int:
mapped = []
killed = 0
for i, v in enumerate(zip(dist, speed)):
mapped.append([v[0]/v[1], v[0], v[1]])
mapped.sort(key=lambda x: x[0])
iterator = iter(mapped)
while True:
try:
next(iterator)[1] = -1
killed += 1
except StopIteration:
return killed
for v in mapped[killed:]:
v[1] = v[1]-v[2]
if v[1] <= 0:
return killed
И он отлично побежал