【leetcode刷题】1、Two Sum两数之和

突然想重刷下之前刷过的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

猜你喜欢

转载自blog.csdn.net/Iv_zzy/article/details/107911858