题目
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
难度
easy(做出来并不难,但要找一个好一点的解法)
思路
找一个O(n)的解法
- 思路转换:两个数之和=target-> 第二个数 = target-第一个数
- 因为既需要比较第二个数与target-第一个数的大小,又需要找出这个数所在的位置,所使用一个map来存储,key为数组中该项的值
- 遍历数组,如果nums[i]不在map的key中,map.put(nums[i],i);如果在,就返回这两个数的序列
代码
class Solution {
public int[] twoSum(int[] nums, int target) {
int s=nums.length;
Map<Integer, Integer> map = new HashMap<>();
for(int i=0;i<s;i++){
if(map.containsKey(target-nums[i])){
//使用map.get获取找到的那个数的位置
return new int[]{map.get(target-nums[i]),i};
}else{
map.put(nums[i],i);
}
}
return new int[2];
}
}
注意点
- 关于key和value,因为要比较的数的大小,所以nums[i]作为key, i作为value
- map存储的key相当于是当前找不到2个符合要求数的集合