题目大意:给定一个整型数组和一个target(整型),要求在数组中找到两个数,使这两个数之和等于target,题目说明每个数组中只存在一对这样的数,输出这两个数的下标(从1开始),并且下标由小到大输出。
理解:题目给的数组是无序的,如果对其进行排序的话就会打乱元素的下标,所以这种方法不太可行。考虑到数组中可能出现负数和重复的数字,所以采取先将数组元素及其下标的关系存储到HashMap<Integer, ArrayList<Integer>>中,ArrayList中存放的是数组元素对应的下标,如果有多个相同的元素,则只存储下标最小的两个。
实现:
public class Solution {
public int[] twoSum(int[] numbers, int target) {
int len = numbers.length;
if(len == 0) return null;
int[] index = new int[2];
HashMap<Integer, ArrayList<Integer>> map = new HashMap<Integer, ArrayList<Integer>>();
for(int i = 0; i < len; i ++) {
if(map.containsKey(numbers[i])) {
if(map.get(numbers[i]).size() < 2) {
map.get(numbers[i]).add(i+1);
}
} else {
ArrayList<Integer> tmp = new ArrayList<Integer>();
tmp.add(i+1);
map.put(numbers[i], tmp);
}
}
for(int i = 0; i < len; i ++) {
int t = target - numbers[i];
if(map.containsKey(t)) {
if(t == numbers[i] && map.get(t).size() == 1) {
continue;
} else if(t == numbers[i] && map.get(t).size() == 2) {
index[0] = map.get(t).get(0);
index[1] = map.get(t).get(1);
} else {
index[0] = map.get(numbers[i]).get(0);
index[1] = map.get(t).get(0);
}
break;
}
}
return index;
}
}