准备刷一刷LeetCode了。
题目:
代码语言:javascript复制'''
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.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] nums[1] = 2 7 = 9,
return [0, 1].
'''
中文题目:
代码语言:javascript复制'''
给定一个整数数组,返回两个元素的索引,使得这两个元素相加的值等于一个给定的target。
您可以假设每个输入都恰有一个解决方案。
例:
给定nums = [2,7,11,15],target = 9,
因为nums [0] nums [1] = 2 7 = 9,
return [0,1]。
'''
解法一和笔记:
代码语言:javascript复制 1 class Solution(object):
2 def twoSum(self, nums, target):
3 """
4 :type nums: List[int]
5 :type target: int
6 :rtype: List[int]
7 """
8 arr = {} # 定义一个空字典
9 length = len(nums)
10 for i in range(length):
11 if (target - nums[i]) in arr: # 如果target减去当前的元素的值存在于空字典中时,符合题意。
12 return [arr[target - nums[i]], i] # 返回target-nums[i]和i的索引
13 arr[nums[i]] = i # 否则将该元素加入字典arr
14
15 # 该方法accepted了,不知道是否有更优的解法
16 a = Solution().twoSum([3, 2, 4], 6)
17 print(a)