1、两数之和 leetcode

题目描述

describtion.png

思路：返回位置题中用python可以有两种方法，一种是位置遍历，返回i；另一种是用数组方法index。用后一种方法，要返回两个坐标时必须分开求，不然数字相同index只返回先出现的位置。先遍历, 在target减去i，在余下数组中求出另外一值

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in nums:
            j = target - i
            start_index = nums.index(i) + 1
            temp_nums = nums[start_index:]
            if j in temp_nums:
                return nums.index(i),start_index+temp_nums.index(j)
            

注意：这里的nums.index()如果两元素相同，则默认返回第一个。
优化：用hashmap。将数值和其在数组里的位置分别作为dict的key和value，比索引快。

        dict = {}
        for i in range(len(nums)):
            if target-nums[i] not in dict:
                dict[nums[i]] = i
            else:
                return [dict[target-nums[i]],i]