题目
来源:力扣(LeetCode)
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
第一版(简单、直接):
两层循环(与上学时输出乘法口诀算法类似)
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
for (int i = 0; i < nums.length - 1; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
result[0] = i;
result[1] = j;
break;
}
}
}
return result;
}
}
结果:
执行用时:85 ms (战胜20%多,作为有追求的人肯定不满足)
内存消耗:39.7 MB
第二版(想办法减少循环层数):
一次循环+用map记录已循环的值及下标
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
Map<Integer, Integer> map = new HashMap<Integer, Integer>(nums.length);
int length = nums.length;
Integer index;
for (int i = 0; i < length; i++) {
index = map.get(Integer.valueOf(target - nums[i]));
if (index != null) {
result[0] = index;
result[1] = i;
break;
} else {
map.put(nums[i], i);
}
}
return result;
}
}
结果:
执行用时:2 ms(战胜99.65%,很好奇剩下0.35%的人怎么实现的)
内存消耗:40 MB
有更好解决方案的朋友可以评论交流学习。