题目
给一个整数数组,找到两个数使得他们的和等于一个给定的数target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是1到n,不是以0开头。
样例
numbers=[2, 7, 11, 15], target=9
return [1, 2]
注意
你可以假设只有一组答案。
解题
更新
更改为HashMap存储
num 不在map中,put( target - num,i)
否则,result[0] = map.get(num)+1 result[1] = i + 1
public class Solution {
/*
* @param numbers : An array of Integer
* @param target : target = numbers[index1] + numbers[index2]
* @return : [index1 + 1, index2 + 1] (index1 < index2)
*/
public int[] twoSum(int[] numbers, int target) {
// write your code here
int[] result = new int[2];
if(numbers ==null || numbers.length <=1){
return result;
}
HashMap<Integer,int[]> map = new HashMap<Integer,int[]>();
for(int i=0;i<numbers.length;i++){
int num = numbers[i];
int[] value = map.get(num);
if(value == null){
value = new int[]{i};
map.put(target - num,value);
}else{
result[0] = value[0] + 1;
result[1] = i + 1;
}
}
return result;
}
}
题目之前做LeetCode时候写过这个,利用这里利用ArrayList,当 target- numbers[i] 不在 list中,把numbers[i] 加入到list中,当存在的时候,取出下标,和当前数的下标就是答案了。
public class Solution {
/*
* @param numbers : An array of Integer
* @param target : target = numbers[index1] + numbers[index2]
* @return : [index1 + 1, index2 + 1] (index1 < index2)
*/
public int[] twoSum(int[] numbers, int target) {
// write your code here
int result[] = new int[2];
ArrayList<Integer> list = new ArrayList<Integer>();
for(int i = 0;i<numbers.length ; i++){
int tmp = target - numbers[i];
if(!list.contains(tmp)){
list.add(numbers[i]);
}else{
result[0] = list.indexOf(tmp) + 1;
result[1] = i + 1;
break;
}
}
return result;
}
}
Java Code
总耗时: 1042 ms
class Solution:
"""
@param numbers : An array of Integer
@param target : target = numbers[index1] + numbers[index2]
@return : [index1 + 1, index2 + 1] (index1 < index2)
"""
def twoSum(self, numbers, target):
# write your code here
L = {}
result = [0]*2
for i in range(len(numbers)):
tmp = target - numbers[i]
if tmp not in L:
L[ numbers[i] ] = i + 1
else:
result[0] = L[tmp]
result[1] = i + 1
break
return result
Python Code
总耗时: 313 ms