LeetCode: 01_two-sum

问题描述:

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

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

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

暴力答案:

基本思路:

嵌套循环,用最笨的方法实现计算。减少了内存,但是严重浪费了时间。

 1 class Solution(object):
 2     def twoSum(self, nums, target):
 3         length=len(nums)
 4         for i in range(0,length-1):
 5             for j in range(i+1,length):
 6                 if nums[i]+nums[j]==target:
 7                     return i,j
 8                     break
 9                 else:
10                     continue

最优答案:

(哈希(Python 叫字典)时间复杂度:O(n)O(n)空间复杂度:O(n)O(n)

一边判断一边创建字典,直到差值也在字典中,返回当前以及另外一个数的下标

 1 class Solution:
 2     def twoSum(self,nums,target):
 3         d = {}  #创建字典
 4         n = len(nums)
 5         for x in range(n):
 6             if target - nums[x] in d: #判断前面另外一个数是否在字典中
 7                 return d[target-nums[x]],x   #返回另外一个数的下标,以及当前数字的下标
 8             else:
 9                 d[nums[x]] = x  #字典赋值
10 
11 作者:chanpin
12 链接:https://leetcode-cn.com/problems/two-sum/solution/wo-shi-pythoncai-ji-liang-shu-zhi-he-de-3ge-jie-fa/
13 来源:力扣(LeetCode)

猜你喜欢

转载自www.cnblogs.com/yueluhun/p/12920089.html