前言
你好, 我是Cici。我一直觉得自己的算法比较菜,所以打算自己重新系统的刷一刷,谈谈自己的思路,说说自己的思考,加强自己的同时也希望能给大家带来帮助。其实数据结构和算法属于我们技术人员的内功,不管技术怎么更新,它始终都是不变的。
一、问题描述
给定一个整数数组 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]
二、题目分析
方法一:暴力枚举
枚举数组中的每个数 x,在数组中寻找是否存在 target - x,如果存在则直接返回。由于答案中不能出现同一元素,所以只需要在 x 的后面寻找 target - x 即可。
时间复杂度:O(n²),n为数组中元素的个数。
方法二:利用哈希表
在方法一中,寻找 target - x 的操作的时间复杂度很高。我们需要快速在数组中寻找目标元素并返回其索引,元素的值和它的索引存在对应关系,所以使用哈希表,key 为元素的值,value 为元素的索引,这样只需要遍历一遍数组,如果哈希表中存在值为 target - x 的 key,则返回 target - x 和 x 的索引即可,否则将 x 存入哈希表,有效避免了重复匹配。
三、参考代码
利用哈希表
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < nums.length; i++){
if(map.containsKey(target - nums[i])){
return new int[]{i, map.get(target - nums[i])};
}else{
map.put(nums[i], i);
}
}
return new int[0];
}
}