描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标
例子
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
- 暴力法:两个for循环
- 增强法(推荐, O(n) 因为要求返回其下标,比较方便):使用字典,查找是O(1)
{需要的数: 我是谁} 遍历每个数,如果我不是被需要的,我就去把我需要谁写在字典里
- 第三种方法(O(nlogn):先保存每个数字的下标,再排序O(nlogn),使用两个指针(需要先排序才能知道,和大于target或小于target时移动哪个指针)
答案 - python
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
# {需要的差值:index}
d={}
for i in range(len(nums)):
if d.get(nums[i]) is not None: #第二个参数不写,默认返回None,注意返回0时,也会被当作false
return [d.get(nums[i]), i]#i会比较大
else:
d[target-nums[i]]=i
return []
- c++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> m;
for (int i=0; i<nums.size(); i++)
if (m.find(nums[i])!=m.end())
return {m[nums[i]], i};
else
m[target-nums[i]]=i;
return {};
}
};