给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
一、Python代码
思路一
按直接暴力地做:
def twoSum(nums, target):
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[i] + nums[j] == target:
return [i,j]
思路2
遍历数组,每次将target - 数组中的值存入字典中,一旦字典中出现重复就发现了数组中和为目标值的两个数。典型地用空间换时间的思路。
def twoSum(nums, target):
dict = {}
for i in range(len(nums)):
if target-nums[i] not in dict:
dict[nums[i]]=i
else:
return [dict[target-nums[i]],i]
时空复杂度
思路一:时间复杂度O(n),空间复杂度O(n)
思路二:时间复杂度O(n),空间复杂度O(1)
二、JAVA代码
思路一
遍历每个元素 xx,并查找是否存在一个值与 target - xtarget−x 相等的目标元素,也就是思路一种加和找目标的思路
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] == target - nums[i]) {
return new int[] { i, j };
}
}
}
throw new IllegalArgumentException("No two sum solution");
}
思路二
一遍哈希表,在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement)) {
return new int[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
网友评论