Python_百题大冲关_7.查找两个总和为特定值的索引_找到两个总和为特定值的索引 python-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_45568867/article/details/115751811

本文介绍了如何高效地找出数组中两数之和等于特定值的索引。首先，列举了暴力求解的O(n²)方法，然后提出使用list内置函数的解决方案，最后详细阐述了利用哈希表实现的O(n)时间复杂度的方法。哈希表的使用大大提高了查找效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

7.查找两个总和为特定值的索引

题目描述

给定一个数组，找到两个总和为特定值的索引。
例如给定数组 [1, 2, 3, -2, 5, 7]，给定总和 7，则返回索引 [1, 4]。

思路一

O(n²)暴力

代码

class Solution(object):

    def two_sum(self, nums, val):
        if nums is None or val is None:
            raise TypeError('nums or target cannot be None')
        if not nums:
            raise ValueError('nums cannot be empty')
        lenth = len(nums)
        res = []
        for i in range(lenth):
            temp = val - nums[i]
            for j in range(lenth):
                if nums[j] == temp:
                    res.append(i)
                    res.append(j)
                    return res
        return None

思路二

使用list内置函数 O(n²)

代码

class Solution(object):

    def two_sum(self, nums, val):
        if nums is None or val is None:
            raise TypeError('nums or target cannot be None')
        if not nums:
            raise ValueError('nums cannot be empty')
        lenth = len(nums)
        res = []
        for i in range(lenth):
            temp = val - nums[i]
            if temp in nums:
                res.append(i)
                res.append(nums.index(temp))
                return res
        return None

思路三

哈希表

代码

class Solution(object):

    def two_sum(self, nums, val):
        if nums is None or val is None:
            raise TypeError('nums or target cannot be None')
        if not nums:
            raise ValueError('nums cannot be empty')
        cache = {}
        for index, num in enumerate(nums):
            cache_val = val - num
            if num in cache:
                return [cache[num], index]
            else:
                cache[cache_val] = index
        return None