【LeetCode练习】 -- 1. 两数之和

189 阅读2分钟

小知识,大挑战!本文正在参与「程序员必备小知识」创作活动

本文已参与「掘力星计划」,赢取创作大礼包,挑战创作激励金。

题目描述

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

解题思路

利用Map的key-value结构,将当前值所需的值和当前值的下标存储在map中,在循环数组的过程中,一边判断当前值时候是其他值所需的,一边将当前值所需的数值和下标放入map

构造Map集合,Map的key为目标值target - 当前值, value为当前值的下标。这样在循环中可以先判断当前值是否可以和前面的值构成目标值target。如何当前值可以和前面的值构成目标值target,那么Map中那个值的下标和当前值的下标就是要返回的答案。

示例代码

    public int[] twoSum(int[] nums, int target) {
        // key: 当前值需要的数值(target - 当前值)
        // value: 当前值的下标
        Map<Integer, Integer> map = new HashMap<>();
        int len = nums.length;
        for (int i = 0; i < len; i++) {
            // 判断当前值是否是前面所需要的
            // 如果map可以get到值,说明当前值是是前面所需要的
            Integer surplus = map.get(nums[i]);
            if (surplus != null && i != surplus){
                return new int[]{i,  surplus};
            }
            // 如果当前值不是前面所需要的,那么把当前值需要的数值和下标放入map
            map.put(target - nums[i], i);
        }

        return null;
    }

LeetCode运行结果 image.png