LeetCode - 001_两数之和

题目描述

给定一个整数数组nums和一个整数目标值target，请你在该数组中找出和为目标值的那两个 整数，并返回它们的数组下标。

示例

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

思路

1. 定义一个map，key为数组的值，value为值对应的索引
2. 遍历数组，判断目标值减当前索引处值是否在map中存在的
3. 如果存在，直接返回相减后值的对应索引和当前索引
4. 不存在，则把当前值和对应索引添加到map里面

Code

public class Solution {
    public int[] twoSum(int[] nums, int target) {
        if (ArrayUtils.isEmpty(nums)) {
            throw new IllegalArgumentException("数组为空");
        }
        Map<Integer, Integer> map = Maps.newHashMap();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(target - nums[i])) {
                return new int[]{map.get(target - nums[i]), i};
            }
            map.put(nums[i], i);
        }
        throw new IllegalArgumentException("数组中没有符合条件的两个数");
    }
}