leetcode系列之两数之和

285 阅读1分钟

1.题目描述

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

2.示例

示例 1:

输入:nums = [2,7,11,15], target = 9 
输出:[0,1] 
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1]

示例 2:

输入:nums = [3,2,4], target = 6 
输出:[1,2] 

示例 3:

输入:nums = [3,3], target = 6 
输出:[0,1]

提示:

`2 <= nums.length <= 104 -109 <= nums[i] <= 109 -109 <= target <= 109` 

**进阶:

你可以想出一个时间复杂度小于 O(n2) 的算法吗?

3.题解

3,1 双重for循环

思路分析

最开始的想法就是直接暴力破解,双重for循环安排上。外层循环每取出一个数字,然后内层循环从该元素位置后面的剩余元素依次取出来与外层循环取出的数字求和,看是否等于目标值 target。直到等于目标值 target时,返回外层循环的数字和内层循环的数字的索引

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for index1 in range(len(nums)):
            for index2 in range(index1+1,len(nums)):
                if int(nums[index1]) + int(nums[index2]) == target:
                    return [index1,index2]

result:

image-20220528195127506.png

3.2 单层循环

思路分析

从3.1的提交结果可以看出,3.1的做法耗时很长,同时也不满足进阶的要求,所以后面又做了一些改进。与之前的做法一样,仍然从nums 每取出一个数字,只是这里不再进行内层循环。而是使用目标值 target减去取出来的数字,判断差值是否也在nums 中,若在nums中,且索引和最开始取出的数字索引不同,则返回二者的索引

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        target = int(target)
        length = len(nums)
        for index1 in range(length):
            other = target - nums[index1]
            if other in nums:
                index2 = nums.index(other)
                if index1 != index2:
                    return [index1,index2]

result:

image-20220528195225576.png