一、题目
题目来源
给定一个整数数组 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) 的算法吗?
二、java代码
垃圾的代码(双层for循环)
思路:给一个数组,然后通过循环来查找符合的数。时间复杂度为O(n^2)
class Solution { public int[] twoSum(int[] nums, int target) { int[] result = new int[2];//创建数组 for(int i = 0; i < nums.length; i++){ //第一层找一个数 int a = nums[i]; for(int j = 0; j < nums.length; j++){ //第二层找第二个数 if(i != j){ //要求两个数不在同一个位置 int b = nums[j]; if((a + b) == target){ result[0] = j; result[1] = i; } } } } return result; } }
在这里很明显可以看到这里有那么一丢丢拉跨,而且逻辑还欠缺那么一丢丢。
好点的代码(使用Map)
使用 map 来解决,建立一一对应的 k-v 键值对,key 存储数值,value 存储数值的下标,时间复杂度为 O(n)
class Solution { public int[] twoSum(int[] nums, int target) { int[] result = new int[2]; if(nums == null || nums.length == 0){ return result; } Map<Integer, Integer> map = new HashMap<>(); for(int i = 0; i < nums.length; i++){ int temp = target - nums[i]; //map中的containsKey(key)方法是判断该key在map中是否有key存在 if(map.containsKey(temp)){ //通过map.get(value)获得对应的位置 result[0] = map.get(temp); result[1] = i; } map.put(nums[i], i); } return result; } }