文章目录
- 1 题目描述
- 2 解题思路
- 2.1 暴力破解
- 2.2 使用 Map
1 题目描述
给定一个整数数组 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 解题思路
2.1 暴力破解
两层 for 循环, 外层循环 0 ~ nums.length - 1
, 内层循环 外层 + 1 ~ nums.length
, 这样虽然能简单粗暴解题, 但时间复杂度 O( n 2 n^2 n2)
2.2 使用 Map
- 目标是
nums[i] + nums[j] = target
, 转换一下变成target - nums[i] = nums[j]
- 在遍历
nums
时, 将数据依次存储到 map, key为nums[i]
, value 为位置 i - 使用
target - nums[i]
的结果去 map 中查找, 看是否能匹配上, 这样就能快速找到nums[j]
时间复杂度为O(n)
class Solution {public int[] twoSum(int[] nums, int target) { int[] result = new int[2]; // 如果数组为空或者长度小于2, 直接返回 if (nums == null || nums.length < 2) { return result; } // key为数组元素, value为数组下标, 保存已经遍历过的元素 Map<Integer, Integer> dataMap = new HashMap<>(nums.length); for (int i = 0; i < nums.length; i++) { // 如果 map 中包含target - nums[i], 则说明找到了 if (dataMap.containsKey(target - nums[i])) { result[0] = dataMap.get(target - nums[i]); result[1] = i; break; } // 如果 map 中不包含target - nums[i], 则将当前元素放入map中 dataMap.put(nums[i], i); } return result; }
}