题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
思路:两次遍历: * 1、用一个 Map<Integer,List<Integer> 存放数组中每个元素的的值与下标,即 (num,[index1,index2]),因为数组中可能有多个值相同的元素 * 2、遍历数组,对指定元素num,看是否存在 target - num 的元素,并获取其下标值 * 1)如果 键为 target - num 对应的 链表中存放的下标数大于等于2 ,则说数组中至少有2个元素的值均为 target/2 ,返回下标即可 比如[3,3,4,5,4] * * 6 * 2)如果 键为 target - num 对应的 链表中存放的下标数为1 ,则要判断存放在链表中的下标值是否等于当前元素num 的下标,为了防止 当num == taeget/2 且元素之出现一次的情况 * 比如[3,4,5,1] * 6
public int[] twoSum(int[] nums, int target) {
Map<Integer,List<Integer>> map = new HashMap<>();
//第一次遍历,记录下数组中元素与其下标
for(int i=0;i<nums.length;i++){
if(map.containsKey(nums[i])){
map.get(nums[i]).add(i);
}else{
List<Integer> list = new ArrayList<>();
list.add(i);
map.put(nums[i],list);
}
}
int i=0;
int j=0;
int[] result = new int[2];
//第二次遍历,找到与当前元素 num 之和为target的另外一个元素的下标
for(;i<nums.length;i++){
if(map.containsKey(target-nums[i])){
List<Integer> list = map.get(target-nums[i]);
if(list.size()>=2){ //如果数组中有两个及以上的元素 num == target/2 ,则直接获取其下标即可
result = new int[]{list.get(0), list.get(1)};
break;
}else {
j = list.get(0);
if(i==j){ //为了应对 [3,4,5,1] 6 情况
continue;
}
result = new int[]{i, j};
break;
}
}
}
return result;
}