文章目录
- 需求
- 暴力求解
- 优化一下暴力解法
- 用Map
- 结尾
需求
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案,并且你不能使用两次相同的元素。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2) 的算法吗?
暴力求解
不就是找到两个值吗, 让其相加等于 target, 两层 for 循环不就搞定了?
public int[] twoSum(int[] nums, int target) {int length = nums.length;int[] res = new int[2];for( int i = 0; i < length; i++ ){for( int j = i + 1; j < length; j++ ){if( i != j && nums[i] + nums[j] == target ){res[0] = i;res[1] = j;return res; }}}return res;
}
执行结果, 时间有点害怕啊…
优化一下暴力解法
提示里说, 2 <= nums.length <= 104
只会存在一个有效答案
,那当数组长度是2直接返回 岂不美哉
public int[] twoSum(int[] nums, int target) {int length = nums.length;int[] res = new int[2];if( length == 2 ){res[0] = 0;res[1] = 1;return res; }for( int i = 0; i < length; i++ ){for( int j = i + 1; j < length; j++ ){if( i != j && nums[i] + nums[j] == target ){res[0] = i;res[1] = j;return res; }}}return res;
}
运行了下 怎么还变慢了~~~
用Map
我们可以一边遍历数组, 一边将每个值与 target 做运算. 比如 target 是7, 数组中的第一个元素是 3 , 那么就可以将 target - 3 的值存进到 Map 里去, key 是 target-3, value是当前数组的下标.
这样我们就可以在遍历的时候一遍存,一边取出来作比较, 满足返回下标即可.
public int[] twoSum(int[] nums, int target) {int length = nums.length;int[] res = new int[2];Map<Integer, Integer> map = new HashMap(length);for( int i = 0; i < length; i++ ){int num = nums[i];if( map.containsKey(num) ){res[0] = map.get(num);res[1] = i;return res;}map.put(target - num, i);}return res;
}
运行结果, 很丝滑~~~
结尾
以上 是我对这道算法的一些遐想和延伸, 可能不是最优解, 但是算法的优化嘛 本身就是一个思索的过程, 能在这个思索和迭代的过程中有所收获和乐趣就是在成长了, 欢迎大家一起来交流更多的解答…