lintcode:两个数的和

题目

两数之和

给一个整数数组，找到两个数使得他们的和等于一个给定的数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