【Leetcode】1：两数之和（Python）

Z时代
2024-01-10
分类：综合

python

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解题思路

直接使用暴力方法，首先找到其中的一个数字，然后在这个数字之后找是否有加起来等于target的数字，没啥难度：

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        i = 0
        # 这里用了两个循环,最好只能使用一个循环
        while i < len(nums):
            j = i + 1
            while j < len(nums):
                if nums[i] + nums[j] == target:
                    return i, j
                j += 1
            i += 1
        return None

方法二：使用排序+双指针法

首先将拿到的列表，进行sort排序，然后使用双指针，左指针指向第一个最小的元素，右指针指向最大的元素，将两个指针指向的数字分别相加，如果相加的数字大小比target的数字更大，那么说明我们需要减少这两个数字之和，右指针向左移动。反之，左指针向右移动，最终终将找到两个和为target的数字，这样一趟下来其时间复杂度只有O(n),比暴力方法所使用的O(n^2)更小。代码如下：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        new_ls=sorted(nums)
        left=0
        right=len(new_ls)-1
        while left < right:
            if (new_ls[left]+new_ls[right]) > target:
                right-=1
            if (new_ls[left]+new_ls[right]) < target:
                left+=1
            if (new_ls[left]+new_ls[right]) == target:
                break
        index_one=nums.index(new_ls[left])
        nums[index_one]=-999999
        index_two=nums.index(new_ls[right])
        return [index_one,index_two]