为什么我在Leetcode中得到的结果与python控制台不同?

luaexgnf  于 2023-04-04  发布在  Python
关注(0)|答案(1)|浏览(212)

Leetcode问题: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,这是错误的答案。

nimxete2

nimxete21#

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

它完美地运行了

相关问题