【Java算法题 16-2】LeetCode - 1.两数之和

82 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第 天,点击查看活动详情


题目、1.两数之和

原题链接:1.两数之和

题目描述

给定一个整数数组 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]

解题思路

题目给出了一个目标数target,要求我们从给定的数组中找出两个数,两数之和等于target,然后还需要输出这两个元素的下标。

咋一看其实不难,完全可以暴力解题,直接给数组双层循环遍历,当找到两数之和为target的元素,返回遍历到的下标就可以。

但是这样耗费的时间太长,时间复杂度不够理想。

我这里的思路,就是使用HashMap集合来存放键值对,主键存放数组元素,Value值存放对应的下标,那么我们就只需要单层的循环就能完成相同效果的遍历,大大节省了时间。

提交代码

class Solution {
    public int[] twoSum(int[] nums, int target) {
        //创建map集合,存放键值对,Key值为数组元素,value值为下标
        Map<Integer,Integer> map = new HashMap<>();

        //遍历数组,将键值对存入集合
        for(int i = 0;i < nums.length;++i){
            map.put(nums[i],i);
        }

        //创建数组,用于存放符合题目要求的下标
        int[] arr = null;

        //再次遍历数组,每遍历一个元素,寻找集合中两者之和为target的元素
        //注意还需要比较两者下标是否相同,相同代表元素重复,不符合要求
        for(int i = 0;i < nums.length;++i){
            if(map.containsKey(target-nums[i]) && map.get(target-nums[i]) != i){
                //找到就记录下标,终止循环
                arr =  new int[]{i , map.get(target-nums[i])};
                break;
            }
        }
        //返回下标数组
        return arr;
    }
}

提交结果

在这里插入图片描述