LeetCode-两数之和

佚名 6年前 (2019-04-25) 随笔 1462人围观抢沙发百度已收录

Question

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

Answer

public void test() {
    int[] nums = new int[]{2, 7, 11, 15};
    int target = 9;
    System.out.println(Arrays.toString(twoSum1(nums, target)));
    System.out.println(Arrays.toString(twoSum2(nums, target)));
}

public int[] twoSum1(int[] nums, int target) {
    for (int i = 0; i < nums.length - 1; i++) {
        for (int j = i+1; j < nums.length; j++) {
            if (nums[i]+nums[j] == target){
                return new int[]{i,j};
            }
        }
    }
    return new int[]{0};
}

public int[] twoSum2(int[] nums, int target) {
    Map<Integer,Integer> map = new HashMap<>(20);
    map.put(nums[0],0);
    for (int i = 1; i < nums.length; i++) {
        if (map.containsKey(target-nums[i])){
            return new int[]{map.get(target-nums[i]),i};
        }
    }
    return new int[]{0};
}