1.题目
给定一个整型数组array和一个整数target,在数组中找出两个整数相加之和为target,返回这两个整数的下标。
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
2.解法
- 最容易想到的方法就是使用两层for循环去遍历,但是这种方法时间复杂度为O(n^2)
public int[] twoSum(int[] array, int target) {
int[] result = new int[2];
for (int i = 0; i < array.length; i++) {
for (int j = i + 1; j < array.length; j++) {
if (array[i] + array[j] == target) {
result[0] = i;
result[1] = j;
}
}
}
return result;
}
- 一种较为优化的解法是使用HashMap,key为当前整型数num,value为下标index,判断target-num是否在当前HashMap中,若在则返回num的下标同时从HashMap中取出target-num的下标,否则将num放入HashMap。
public int[] twoSum(int[] array,int target){
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int i = 0;i<array.length;i++){
int tmp = target-array[i];
if(map.containsKey(tmp)){
return new int[]{i,map.get(tmp)};
}else{
map.put(array[i],i);
}
}
throw new IllegalArgumentException("not find two num");
}
这个问题在后面做了引申,列出来一堆相似的问题,链接:http://bigdataer.net/?p=600