提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
提示:这里可以添加本文要记录的大概内容:
本文只用于记录作者的刷题笔记,极有可能会有错误,欢迎各位批评指正!
提示:以下是本篇文章正文内容,下面案例可供参考
一、题目
示例:pandas 是基于NumPy 的一种工具,该工具是为了解决数据分析任务而创建的。
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
二、思路
首先想到这题可以遍历数组中的元素num,然后判断target-num是否存在来解题,因为需要判断元素是否存在,首先想到哈希表(可以很快的找到一个元素是否存在),此外这题不仅要保存元素,还要保存下标,因此可以用HashMap解决。 作者一开始的做法是把数组中所有元素和元素下标存到HashMap中,但是这样有一个问题,因为数组元素可能会重复,因此不可以用元素值为key,只能以数组元素下标为key,value为数组元素值。但是这样又导致了查找target-num是否存在时只能通过map.containsValue来判断,这相比map.containsKey的时间复杂度高了许多。《代码随想录》中的思路是以数组元素作为Key,元素下标作为Value,不需要先把所有数组元素存到HashMap中,而是直接遍历数组,如果HashMap.containsKey(target-num)则可以直接返回,否则将该数组元素添加到HashMap中。
三、代码实现(JAVA)
1、作者的笨方法
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
int index = 0;
for(int i:nums){
map.put(index,i);
index += 1;
}
for(int key=0;key<nums.length;key++){
int value = nums[key];
map.remove(key);
int sub = target - value;
if(map.containsValue(sub)){
for(Integer newKey:map.keySet()){
if(sub==map.get(newKey)){
return new int[]{key,newKey};
}
}
}
}
return new int[]{0,0};
}
}
2、《代码随想录》中的方法
public int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
if(nums == null || nums.length == 0){
return res;
}
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < nums.length; i++){
int temp = target - nums[i]; // 遍历当前元素,并在map中寻找是否有匹配的key
if(map.containsKey(temp)){
res[1] = i;
res[0] = map.get(temp);
break;
}
map.put(nums[i], i); // 如果没找到匹配对,就把访问过的元素和下标加入到map中
}
return res;
}
总结
1、判断元素是否存在,先想到哈希表,要判断的元素应该作为map的key。