LeetCode167. 两数之和 II - 输入有序数组 + 1. 两数之和

167. 两数之和 II - 输入有序数组（升序数组）

力扣

1.1题目描述

给定一个已按照 升序排列  的有序数组，找到两个数使得它们相加之和等于目标数。

函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。

说明:

• 返回的下标值（index1 和 index2）不是从零开始的。
• 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。

示例:

输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

2代码解析

推荐解法：双指针（缩减搜索空间的思想）

class Solution(object):
    def twoSum(self, numbers, target):
        low = 0
        high = len(numbers) - 1
        while low < high:
            sum = numbers[low] + numbers[high]
            if sum < target:
                low += 1
            elif sum > target:
                high -= 1
            else:
                return [low + 1, high + 1]
        return [-1, -1]

class Solution {
    public int[] twoSum(int[] numbers, int target) {
        int i = 0;
        int j = numbers.length - 1;
        while (i < j && numbers[i] + numbers[j] != target){
            if (numbers[i] + numbers[j] < target){
                i++;
            }else{
                j--;
            }
        }
        return new int[] {i, j};
    }
}

 

• 时间复杂度：O(n)。每个元素最多被访问一次，共有 n 个元素。
• 空间复杂度：O(1)。只是用了两个指针

双指针（缩减搜索空间的思想）同类题

• 11. 盛最多水的容器 双指针-LeetCode11. 盛最多水的容器_IOT_victor的博客-CSDN博客
• 240：搜索二维矩阵II （通用模板） 4：二维数组中的查找（剑指offer第2版）LeetCode240：搜索二维矩阵II_IOT_victor的博客-CSDN博客
• 57-2.和为S的连续正数序列(滑动窗口思想) 57-2：和为S的连续正数序列（剑指offer第2版Python）_IOT_victor的博客-CSDN博客

法二：哈希表（O(N),O(N)）没有用到输入数组已经排序的性质

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        hash_map = dict()
        for index, element in enumerate(numbers):
            # 如果本次索引index值element的数，在已有hashmap中找到对应项
            if target - element in hash_map:
                #思考:index一定大于hash_map[target - element]
                return [hash_map[target - element] + 1 ,index + 1]
            hash_map[element] = index

3其他解法

# two-pointer
def twoSum1(self, numbers, target):
    l, r = 0, len(numbers)-1
    while l < r:
        s = numbers[l] + numbers[r]
        if s == target:
            return [l+1, r+1]
        elif s < target:
            l += 1
        else:
            r -= 1
 
# dictionary           
def twoSum2(self, numbers, target):
    dic = {}
    for i, num in enumerate(numbers):
        if target-num in dic:
            return [dic[target-num]+1, i+1]
        dic[num] = i
 
# binary search        
def twoSum(self, numbers, target):
    for i in xrange(len(numbers)):
        l, r = i+1, len(numbers)-1
        tmp = target - numbers[i]
        while l <= r:
            mid = l + (r-l)//2
            if numbers[mid] == tmp:
                return [i+1, mid+1]
            elif numbers[mid] < tmp:
                l = mid+1
            else:
                r = mid-1

变形：返回所有的多个解

# 有序，多解
nums = [1, 3, 5, 7, 9, 15]
n = 8
print(func(nums, n))  # [(1, 7), (3, 5)]

def func(nums, n):
    low = 0
    high = len(nums) - 1
    ret_list = []
    while low < high:
        sum = nums[low] + nums[high]
        if sum < n:
            low += 1
        elif sum > n:
            high -= 1
        else:
            ret_list.append((nums[low], nums[high]))
            low += 1
            high -= 1
    return ret_list

1. 两数之和（哈希表）

力扣

2.1题目描述

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

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

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

2.2代码详解

推荐写法：遍历列表同时查字典，python用字典模拟哈希表查询

1. key in dict 用时 1.088 秒
2. dict.get(key) 用时 1.294 秒
3. dict[key] 用时 1.01 秒

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        hashmap = {}
        for i, number in enumerate(nums):
            if target - number in hashmap:
                # if hashmap.get(target - number) is not None:
                return [hashmap[target - number], i]

            hashmap[number] = i  # 当碰到相同的元素时，hashmap[number] = i 会将最近遍历的元素索引写入字典
            # 这句不能放在if语句之前，解决list中有重复值或target-num=num的情况

• 时间复杂度：O(n)， 只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找只花费 O(1) 的时间。  
• 空间复杂度：O(n)， 额外空间取决于哈希表中存储的元素数量，该表最多需要存储 n 个元素

其他写法

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dic = {}
        for i, num in enumerate(nums):
            if num in dic:
                return [dic[num], i]
            else:
                dic[target - num] = i
                
        # """这样写更直观，遍历列表同时查字典"""
        # dct = {}
        # for i, n in enumerate(nums):
        #     if target - n in dct:
        #         return [dct[target - n], i]
        #     dct[n] = i