两数之和牛客网 python

最新推荐文章于 2024-07-18 13:30:43 发布

原创最新推荐文章于 2024-07-18 13:30:43 发布 · 246 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #python

该博客讨论了一种解决寻找数组中两个数之和等于目标值问题的方法。通过建立哈希映射，可以在O(n)的时间复杂度内找到满足条件的数对下标。示例展示了对于数组[3, 2, 4]和目标值6，如何返回下标为[2, 3]的结果。提供的代码实现了这一解决方案。

描述

给出一个整数数组，请在数组中找出两个加起来等于目标值的数，你给出的函数twoSum 需要返回这两个数字的下标（index1，index2），需要满足 index1 小于index2.。注意：下标是从1开始的假设给出的数组中只存在唯一解.

例如：

给出的数组为 {20, 70, 110, 150},目标值为90
输出 index1=1, index2=2

示例1

输入：[3,2,4],6

返回值：[2,3]

说明：因为 2+4=6 ，而 2的下标为2 ， 4的下标为3 ，又因为下标2 < 下标3 ，所以输出[2,3]

代码：

class Solution:
    def twoSum(self , numbers , target ):
        result = [0, 0]
        mp = {}
        for i in range(len(numbers)):
            mp[numbers[i]] = i
            for i in range(len(numbers)):
                if target - numbers[i] in mp:
                    if mp[target - numbers[i]] != i:
                        result[0] = i + 1
                        result[1] = mp[target - numbers[i]] + 1
                        return sorted(result)
        return sorted(result)