【力扣-Python-1】两数之和(easy)

https://leetcode.cn/problems/two-sum/

题目描述

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

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

你可以按任意顺序返回答案。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
【力扣-Python-1】两数之和(easy)_第1张图片

解题思路及代码

解法一:两重for循环

思路:先拿出列表里的第0个数字,依次尝试和第1个、第2个……第n个相加,看能否等于目标数字,如果有某个组合等于目标数字,就返回这个组合的两个索引值,如果都不行,再拿出第1个数字,依次尝试和第2个、第3个……第n个相加,看能否等于目标数字……每个数字都不需要回过头和前面的数字相加,因为之前已经尝试过了

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            for j in range(i+1, len(nums)):
                if nums[j] == target - nums[i]:
                    return [i, j]
                else:
                    return "No Exists"

两重循环,时间复杂度就是O(n^2),其中n=len(nums)

解法二:单重for循环+字典存储值(key)和索引(value)

思路:先建立一个用于存放数字(key)和索引(value)的空字典。从列表中取出一个数字,然后看字典里是否存在能跟这个数字相加得到目标数字的数字。如果存在,就返回两个数字的索引,不存在,就把取出来的这个数字作为键,它的索引作为值存进字典,并取出下一个数字,重复刚才的操作,直到找到合适的两个数字或者全部数字都看过一遍为止

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dict = {}  # 定义一个空字典
        for i, value1 in enumerate(nums):
            value2 = target - value1
            if dict.get(value2) != None:  # value2作为字典的key
                return [dict[value2], i]  # dict[value2]为该值对应的索引,即数组下标
            dict[value1] = i  # 若没找到,就把这个值和下标存进字典

        else:
            return "No Exists"

扫描一次就能得到结果,时间复杂度就是O(n),其中n=len(nums)


知识点总结

(一)enumerate()函数

用法:用来遍历一个集合对象,它在遍历的同时还可以得到当前元素的索引位置

names = ["Alice","Bob","Carl"]
for index,value in enumerate(names):
    print(f'{index}: {value}')

运行结果:

0: Alice
1: Bob
2: Carl

(二)dict.get()

用法:get()函数用于获取字典中对应key的value。第一个参数即为key,第二个参数设置的是字典中不包含该键时,get()函数的返回值。若不设置第二个参数,则默认返回None

d = {'a':2, 'b': 3}

d.get('a')
d.get('c')
d.get('f', 5)

运行结果:

2
None
5

你可能感兴趣的:(LeetCode,算法,数据结构,python)