Leetcode1–两数之和
Leetcode1:两数之和
编程语言:python
题目描述
原题链接:https://leetcode-cn.com/problems/two-sum/ (中文)
https://leetcode.com/problems/two-sum/ (英文)
题目描述:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解题思路
方法1:
使用暴力求解的方法(Leetcode提交,可能会超时)
def twoSum(nums, target):
for i in range(len(nums)-1):
for j in range(i+1, len(nums)): #这里下标要从i+1开始,避免重复利用数组中的数
if target == nums[i]+nums[j]:
return i,j
return []
方法2:
使用哈希表建立元素值和索引之间的对应关系
方法1使用的方法的时间复杂度为
O
(
n
2
)
O(n^2)
O(n2),如果已知哈希表,去寻找元素的对应下标(或者已知下标寻找元素)时间复杂度为
O
(
1
)
O(1)
O(1),所以使用哈希表的方法,可以提升时间效率(损失的是空间,以空间换时间)。
def twoSum(nums, target):
length = len(nums)
#初始化一个空哈希表
hash_dict = {}
#遍历数组中的元素
for i in range(length):
# 判断当前值是否在字典中,若是存在返回字典中元素对应的下标和当前的i
# 若是不存在,计算另一个元素的值,把该值作为字典的key,当前循环的i作为value
# 这样存储的方式是,即遍历的过程中就可以直接返回当前下标和另一个对应元素的下标
if nums[i] in target_dict:
return hash_dict[nums[i]], i
else:
tmp = target - nums[i]
hash_dict[tmp] = i
return []
同样的方法,另外一种方式code
def twoSum(nums, target):
hash_dict = {}
for index, value in enumerate(nums):
if value in hash_dict:
return hash_dict[value], index
hash_dict[target-value] = index
return []
C++解题
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
vector<int> res;
for (int i=0; i<nums.size(); i++) {
if(map.find(nums[i]) == map.end()) {
map[target-nums[i]] = i;
}
else {
res.push_back(i);
res.push_back(map[nums[i]]);
break;
}
}
return res;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分析技术问题,我们一起学习进步