LeetCode-N1-两数之和

原题链接

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例：

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

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

从题目上看，最简单的解决方案就是两层for循环。

但这里有一个优化的思路。那就是用空间换取时间。用一个Map来存储遍历过的值和它的索引。这样就只需要一层for循环遍历就可以找到满足需求的数据。

public class N1 {
    public int[] twoSum(int[] nums, int target) {
        // key为值，value为索引
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            Integer res = map.get(target - nums[i]);
            if (res != null)
                return new int[]{res, i};
            else map.put(nums[i], i);
        }
        return new int[]{};
    }
}