python -- leetcode 刷题之路

第一题

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

通用写法 vs 列表推导式, 看到 leetcode 上的 耗时 时快时慢，也是茫然。。。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        for i,v in enumerate(nums):
            for j in range(i+1,len(nums)):
                if v+nums[j] == target:
                    return [i,j]
        else:
            raise ValueError('未找到这样的数字')
            
        # return [[i,j] for i,v in enumerate(nums) for j in range(i+1,len(nums))  if v+nums[j] == target ]
                    
nums = [2, 7, 11, 15]
target = 9
s = Solution()
s.twoSum(nums,target)


# 列表推导式
class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        return [[i,j] for i,v in enumerate(nums) for j in range(i+1,len(nums))  if v+nums[j] == target ]
                    
nums = [2, 7, 11, 15]
target = 9
s = Solution()
s.twoSum(nums,target)