LeetCode-1-两数之和

发表于 更新于 分类于 ，

题目描述

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

示例:

1
2
3
4
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

我的思路

One

  1. 我能想到的只有一个循环去遍历每个数字，一个循环去匹配是否有符合的———— $ O(n^2) $
  2. 首先想到暴力破解，两个for循环去找哪两个索引对应的值加起来等于target(当然这估计过不了关)
  3. 但是想到了HashMap有个get方法效率很高，那么首先把数组中所有元素put到map中，将target - nums[i]作为key，索引index作为value，那么再写一个循环去遍历数组时候，判断起来就很方便了。有关HashMap的一些介绍请移步我对HashMap的理解
  4. 直接用map.get(nums[i])，即可拿到对应的value，当value不为null的时候证明满足了nums[i] + nums[j] = target,也就是nums[i] = target - nums[j]。如上面示例的话，map中会存放的是map:{7:0, 2:1, -2:2, -6:3}
  5. 然后再写一个循环去判断 int b = map.get(nums[i])不为空 且b != i时(因为当nums[0]==3，target==6就会得到[0,0]，很显然这是错误的，题目中提到了不能重复利用这个数组中同样的元素)，即可break，[i,b]就是答案

Two

  1. 不过还是写个两个循环，能否写成一个循环呢？（根据提交的答案结果看来，两者差距不大）
  2. 那就是一个循环，边循环，边put到map中（key,value同上），循环过程中再去判断是否map已经contains这个nums[i]了，如果包含则满足了nums[i] = target - nums[j]，即可以break

上面都将 $ O(n^2)->O(n) $

我的代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package com.dasnnj.practice.simple;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * Description <p> TODO:
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
 * <p>
 * 示例:
 * <p>
 * 给定 nums = [2, 7, 11, 15], target = 9
 * <p>
 * 因为 nums[0] + nums[1] = 2 + 7 = 9
 * 所以返回 [0, 1]</p>
 *
 * @author DASNNJ <a href="mailto:dasnnj@outlook.com"> dasnnj@outlook.com </a>
 * @date 2019-04-28 15:38
 */
public class TwoSum {
    static int[] twoSum(int[] nums, int target) {
        //One
        /*//key为target-nums[i],value为index
        Map<Integer, Integer> map = new HashMap<>();
         for (int i = 0; i < nums.length; i++) {
            map.put(target-nums[i],i);
         }
        int[] res = new int[2];
        for (int i = 0; i < nums.length; i++) {
            Integer b;
            if ((b = map.get(nums[i])) != null && b != i) {
                res[0] = b;
                res[1] = i;
                break;
            }
        }
        return res;
        */
        //Two
        //key为target-nums[i],value为index
        Map<Integer, Integer> map = new HashMap<>();
        int[] res = new int[2];
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i])) {
                res[0] = map.get(nums[i]);
                res[1] = i;
                break;
            }
            map.put(target - nums[i], i);
        }
        return res;
    }

    public static void main(String[] args) {
        int[] nums = new int[]{2, 7, 11, 15};
        System.out.println(Arrays.toString(TwoSum.twoSum(nums, 9)));
    }
}