Python教程

LeetCode刷题Python3之1、两数之和

本文主要是介绍LeetCode刷题Python3之1、两数之和,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

1、两数之和

题目描述

	给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
	你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
	你可以按任意顺序返回答案。

示例 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 <= 103
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109 只会存在一个有效答案

解法一

核心思想:使用两个for循环来遍历该数组,数组下标不能一样,避免出现重复的可以加上标识符来退出程序。
代码如下:

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        arr=[]
        flag=0
        for i in range(len(nums)):
            for j in range(len(nums)):
                if nums[i]+nums[j]==target and flag!=1 and i !=j:
                    arr.append(i)
                    arr.append(j)
                    flag=1
        return arr

这个就是常规求法(暴力枚举求解),这样时间复杂度为O(n^2),要是对执行时间有要求估计就不行了。

解法二

核心思想:先将数组使用enumerate()函数变成索引序列,将数据保存到字典中,然后遍历求解即可!

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dict1={}
        #uid是下标,num是值
        #用值作为键,不怕冲突
        for uid,num in enumerate(nums):
            dict1[num]=uid
        for uid,num in enumerate(nums):
            #target-num是剩下多少,然后遍历
            j=dict1.get(target-num)
            if uid!=j and j!=None:
                return [uid,j]
        return []

可能看上去有点难懂,用个例题解释一下

输入:
nums=[2,3,4,4,7],target=5

1、先使用enumerate函数,第一个参数为下标,第二个参数为值
所以
下标 值
0 2
1 3
2 4
3 4
4 7

2、将值作为键,下标作为值,生成字典dict1
dict1[2]=0
dict1[3]=1
dict1[4]=2
dict1[4]=3
#这里键重复了,那么只记录后一个的值即dict1[4]=3
dict[7]=4

3、遍历enumerate函数
已知target=5,然后dict1[2]=0,那么可以知道还差值为3的数(target-2),那么我们使用get来获取3的键即可
1=dict1[3]=dict1[5-2]
那么此时下标就为0,1,返回即可!!!

用特例来测试
已知:nums=[3,3],target=6

dict1[3]=0
dict1[3]=1
∴可得出
dict1[3]=1

遍历enumerate函数
uid=0,num=3
j=dict1.get(6-3)=dict1.get(3)
此时j=1

因为uid不等于j且j不为空,所以返回即可!!!

这篇关于LeetCode刷题Python3之1、两数之和的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!