给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
在真实的面试中遇到过这道题?
第一种思路:暴力枚举
暴力法很简单,遍历每个元素 xxx,并查找是否存在一个值与 target−xtarget - xtarget−x 相等的目标元素。
class Solution {
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 };
}
}
}
}
}
复杂度分析:
时间复杂度:O(n^2),
对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费 O(n)O(n)O(n) 的时间。因此时间复杂度为 O(n^2)。
空间复杂度:O(1)。
第二种思路:哈希,我们能否通过索引来快速查找元素呢?类似于通过下标来查找元素,以空间换时间。刚开始打算用book数组来标记,但显然没有HashMap好用,因为HashMap底层是用数组+链表来实现的,而且不需要我们自己手写代码来解决冲突,何乐不为呢。JDK1.8好像引入了红黑树。
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int i = 0;i < nums.length;++i)
{
map.put(nums[i],i);
}
for(int i = 0;i < nums.length;++i)
{
int ans = target-nums[i];
if(map.containsKey(ans) && map.get(ans) != i)
{
return new int[]{i,map.get(ans)};
}
}
return new int[2];
}
}