leetcode题目1：两数之和（java）

题目描述

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

示例

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

解题思路

1. 定义一个map
2. 对数组进行遍历，遍历的数组值为A, 判断map的key中是否存在以(target-A)的元素，如果存在则返回。如果不能存在就将(target-A)放进map中。如此一次遍历即可得到结果。

代码

public int[] twoSumV2(int[] nums, int target) {
    if (nums == null || nums.length < 2) {
        return null;
    }
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        if (map.containsKey(nums[i]) && map.get(nums[i]) != i) {
            return new int[]{map.get(nums[i]), i,};
        }
        map.put(target - nums[i], i);
    }
    return null;
}