leetcode 1. 两数之和(Java版)

题目描述:(题目难度,简单)

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

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

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

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

示例代码

下面给出时间复杂度为,空间复杂度为的Java代码(可以直接在 leetcode 上提交):

class Solution {
    public int[] twoSum(int[] nums, int target) {//前两行是 leetcode 提供的模板
        int[] res = {0,0};
        HashMap hashMap = new HashMap<>();
        for(int i = 0; i < nums.length; ++i){
            if(hashMap.size() > 0){
                if(hashMap.containsKey(target - nums[i])){
                    res[0] = hashMap.get(target - nums[i]);
                    res[1] = i;
                    return res;
                }
            }
            hashMap.put(nums[i], i);
        }
        return res;
    }
}

思路解析

在遍历数组的过程中,将遍历过的元素都扔到 HashMap 当中。当遍历到当前元素时,查找目标值与当前元素的差值是否在 HashMap 中,如果存在,则算法结束,如果不存在,则将当前元素加入到 HashMap 中,然后遍历下一元素。
为什么能这么做?因为如果输入有解的话,那么要找的那两个元素必定一前一后,当我们遍历到后一元素时,前一元素必定在 HashMap 中,即查找一下 HashMap 便可以得到答案。

你可能感兴趣的:(leetcode 1. 两数之和(Java版))