两数之和
题目描述
思路
暴力枚举
最简单的想法就是枚举数组中的每个数字,计算target-x是否存在于数组中,不存在则继续往下枚举,存在返回当前数字下标和target-x的下标即可。
Python实现
# 暴力枚举
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
for i in range(n):
for j in range(i+1, n):
if nums[i] + nums[j] == target:
return [i ,j]
return []
Java实现
/**
* 暴力枚举
*/
class Solution {
public int[] twoSum(int[] nums, int target) {
int n = nums.length;
for (int i = 0; i < n; ++i) {
for (int j = i+1; j < n; ++j) {
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
}
}
return new int[0];
}
}
哈希表
是用哈希表,对于每个x,首先查询哈希表中是否存在target-x,然后将x插入到哈希表中,即可保证不会让x和自己匹配。
Python实现
# 哈希表
class Solution:
def twoSum(self, nums:List[int], target: int) -> List[int]:
map = dict()
for i, num in enumerate(nums):
if target - num in map:
return [map[target-num], i]
map[num] = i
return []
Java实现
/**
* 哈希表
*/
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; ++i) {
if (map.containsKey(target - nums[i])) {
return new int[]{map.get(target-nums[i]), i};
}
map.put(nums[i], i);
}
return new int[0];
}
}