LeetCode刷题第1题:两数之和

原文链接：https://leetcode.cn/problems/two-sum/description/

题意

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

示例

实例1：

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

示例2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例3：

输入：nums = [3,3], target = 6
输出：[0,1]

解题

• 暴力法：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<nums.length;j++){
                if(nums[i]+nums[j]==target){
                    return new int[]{i,j};
                }
            }
        }
        throw new IllegalArgumentException("没找到");
    }
}

• 哈希映射

解题思路

遍历数组 nums，i 为当前下标，每个值都判断map中是否存在 target-nums[i] 的 key 值。如果存在则找到了两个值，如果不存在则将当前的 (nums[i],i) 存入 map 中，继续遍历直到找到为止；如果最终都没有结果，则抛出异常。

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> map = new HashMap<>();
        for(int i=0;i<nums.length;i++){
            int findNum = target - nums[i];
            if(map.containsKey(findNum)){
                return new int []{i,map.get(findNum)};
            }
            map.put(nums[i],i);
        }
        throw new IllegalArgumentException("没找到");
    }
}