美文网首页
LeetCode题解1:Two Sum

LeetCode题解1:Two Sum

作者: 沙札罕 | 来源:发表于2015-06-30 16:54 被阅读261次

Two Sum问题:给定一个数组nums和一个正整数target,试从数组中找出2个元素,它们相加之和恰好为target。

难度:容易

思路:用Hash表建立反向索引(很多查找类问题都可以用索引来优化性能)

陷阱:index1 != index2

代码:

class Solution:
    # @param {integer[]} nums
    # @param {integer} target
    # @return {integer[]}
    def twoSum(self, nums, target):
        nums_dict = {}
        for idx, num in enumerate(nums):
            nums_dict[num] = idx
        for index1, num in enumerate(nums):
            index2 = nums_dict.get(target - num)
            if index2 is not None and index2 != index1:
                return [index1 + 1, index2 + 1]
        return [0, 0]

时间复杂度:O(n)

空间复杂度:O(n)

相关文章

网友评论

      本文标题:LeetCode题解1:Two Sum

      本文链接:https://www.haomeiwen.com/subject/bpoeqttx.html