1. 两数之和

2023-12-13 16:49:45

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

这题比较简单,用O(N^{2})的方法很容易就能解答,现在直接上O(N\log N)的方法。

方法一:?

利用一个元素为元组的列表存储每个数以及对应的索引,然后对该列表排序,该过程是NlogN的复杂度,最后用双指针的思路对列表的元素进行判断即可。

    def twoSum(self, nums: List[int], target: int) -> List[int]:
        if nums[0] + nums[1] == target:
            return [0, 1]
        nums = [(v, i) for i, v in enumerate(nums)]
        nums.sort()
        left, right = 0, len(nums) - 1
        while left < right:
            if nums[left][0] + nums[right][0] == target:
                return [nums[left][1], nums[right][1]]
            elif nums[left][0] + nums[right][0] < target:
                left += 1
            else:
                right -= 1

方法二:?

该方法更牛逼,是O(N)的复杂度,但是有点不好理解有点绕。

    def twoSum(self, nums: List[int], target: int) -> List[int]:
        if nums[0] + nums[1] == target:
            return [0, 1]
        record = {}
        for i, v in enumerate(nums):
            # 要和当前数相加等于target的数为diff,如果后续元素中有diff,那么就是要输出的数
            diff = target - v
            # 如果当前元素在record中,意味着当前元素能和曾经出现过的相加等于target,故输出
            if v in record:
                # 当某个v存在于record中,意味着该v是曾经的某个diff,而当时那个diff对应着当时与diff相加满足条件的那个v,我们在当时记录了那个v的索引,因此这里为record[v]
                return [record[v], i]
            else:
                # 能使和diff相加为target的当前元素的下标为i
                record[diff] = i

文章来源:https://blog.csdn.net/qq_36583400/article/details/134971587
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。