1.two_sum题目

题目

Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

---

思路

直观思路是使用两个for循环遍历nums，然后if判断。

class Solution(object):  
    def two_sum1(self, nums, target):  
        for i in range(0, len(nums)):  
            for j in range(i+1, len(nums)):  
                if((nums[i]+nums[j])==target):  
                    a = i  
                    b = j  
                    return[a, b]

Submit后AC，但是这种暴力搜索的时间复杂度较高。另一种更快速的方法是哈希法。

class Solution:  
    def twoSum(self, nums, target):  
        """ 
        :type nums: List[int] 
        :type target: int 
        :rtype: List[int] 
        """  
        d = {}  
        for i, j in enumerate(nums):  
            if target - j in d:  
                return(d[target-j], i)  
            else:  
                d[j] = i