【LeetCode 1】Two Sum(Python)

来源:互联网 发布:复杂网络 python 编辑:程序博客网 时间:2024/06/05 00:13

Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution.

实例:
输入: numbers={2, 7, 11, 15}, target=9
输出: (0,1)

题目分析:
给定一个整数数组,任意输入一个数,在数组中查找并返回能够相加得到这个数的两个数的位置。不能排序,否则会打乱原始数组内的位置,题目要求就是找到原始位置。

方法一:

  1. 思路:用双重循环,得到任意组合相加的和,判断是否与输入的数相等。

  2. 缺点:时间复杂度过高为O(n²),运行速度大打折扣。而且提交时因为时间复杂度会提示不合格 优点:最简单的思路

  3. 代码:(此代码不能成功提交,原因↑)
class Solution:    def twoSum(self, nums, target):        for i in range(len(nums)):            a=nums[i]            for j in range(len(nums)):                if target-a==nums[j]:                    return i,jt=Solution()arr=[1,2,3,6]jg=t.twoSum(arr,4)print(jg)
运行结果:(0,2)

方法二:

  1. 思路:要解决时间复杂度的问题,就要简化循环,只使用一重循环,在参考了别人的代码后,发现方法就是创建一个空字典,通过循环往进添加元素的同时查找已添加的元素中是否有符合条件的结果。
  2. 优点:时间复杂度为O(n),空间复杂度为O(n)。 缺点:难理解。
  3. 代码:(成功提交)
class Solution:    def twoSum(self, nums, target):        dict={}        for i in range(len(nums)):            x=nums[i]            if target-x in dict:                return (dict[target-x],i)            dict[x]=i   #此时确定了字典内的内容为dict{x:i}注意这里的KEY(其实就是顺序123)放在了i的位置,x的位置放的是数组中元素。这样上一行的dict[target-x]返回的才是位置t=Solution()arr=[3,2,3]jg=t.twoSum(arr,6)print(jg)

总结:用方法二才可以通过。我最开始的思路太简单,没有考虑到时间复杂度的问题。以后在做题过程中一定要考虑运行速度的问题。网上还有好多种解法,冒泡排序,哈希函数之类的,有空要去试着理解一下,运行速度应该比我这种蠢方法好。但我觉得还是我这两种方法好理解。而且作为个初学者应该也还可以吧。

原创粉丝点击