题目
给一个整数数组,找到两个数使得他们的和等于一个给定的数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; } }
总耗时: 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
总耗时: 313 ms