突然想重刷下之前刷过的leetcode的题,记录下自己的解题思路,以及不同的方法
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
Given
nums = [2, 7, 11, 15], target = 9,
Because
nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
解法一:
最普通,常规的解法,像排序算法,时间复杂度是O(n2)
number = len(nums)
for i in range(0,number-1):
for j in range(i+1,number):
obj = nums[i] + nums[j]
if obj == target:
list_result = [i,j]
return list_result
解法二:
类似于折半查找,先需要个排序,时间复杂度是O(nlogn)
sorted_id = sorted(range(len(nums)), key=lambda k: nums[k])
head = 0
tail = len(nums) - 1
sum_result = nums[sorted_id[head]] + nums[sorted_id[tail]]
while sum_result != target:
if sum_result > target:
tail -= 1
elif sum_result < target:
head += 1
sum_result = nums[sorted_id[head]] + nums[sorted_id[tail]]
return [sorted_id[head], sorted_id[tail]]
解法三:
用字典记录每个元素的下标,边记录边查找,省得多次查找,时间复杂度是O(n)。
hashmap = {}
for index,ele in enumerate(nums):
if target - ele in hashmap:
return [hashmap[target - ele],index]
hashmap[ele] = index