力扣: 两数之和梦开始的地方

最新推荐文章于 2024-09-29 16:57:04 发布

长安初雪

最新推荐文章于 2024-09-29 16:57:04 发布

阅读量525

点赞数 12

分类专栏：算法是程序的灵魂文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/Snowyyds/article/details/141387692

版权

算法是程序的灵魂专栏收录该内容

32 篇文章 0 订阅

订阅专栏

文章目录

需求
暴力求解
优化一下暴力解法
用Map
结尾

需求

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

提示：
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

暴力求解

不就是找到两个值吗, 让其相加等于 target, 两层 for 循环不就搞定了?

public int[] twoSum(int[] nums, int target) {
    int length = nums.length;
    int[] res = new int[2];
    for( int i = 0; i < length; i++ ){
        for( int j = i + 1; j < length; j++ ){
            if( i != j && nums[i] + nums[j] == target ){
                res[0] = i;
                res[1] = j;
                return res; 
            }
        }
    }
    return res;
}

执行结果, 时间有点害怕啊…

在这里插入图片描述

优化一下暴力解法

提示里说, 2 <= nums.length <= 104 只会存在一个有效答案,那当数组长度是2直接返回岂不美哉

public int[] twoSum(int[] nums, int target) {
    int length = nums.length;
    int[] res = new int[2];
    if( length == 2 ){
        res[0] = 0;
        res[1] = 1;
        return res; 
    }
    for( int i = 0; i < length; i++ ){
        for( int j = i + 1; j < length; j++ ){
            if( i != j && nums[i] + nums[j] == target ){
                res[0] = i;
                res[1] = j;
                return res; 
            }
        }
    }
    return res;
}

运行了下怎么还变慢了~~~

在这里插入图片描述

用Map

我们可以一边遍历数组, 一边将每个值与 target 做运算. 比如 target 是7, 数组中的第一个元素是 3 , 那么就可以将 target - 3 的值存进到 Map 里去, key 是 target-3, value是当前数组的下标.
这样我们就可以在遍历的时候一遍存,一边取出来作比较, 满足返回下标即可.

public int[] twoSum(int[] nums, int target) {
    int length = nums.length;
    int[] res = new int[2];
    Map<Integer, Integer> map = new HashMap(length);
    for( int i = 0; i < length; i++ ){
        int num = nums[i];
        if( map.containsKey(num) ){
            res[0] = map.get(num);
            res[1] = i;
            return res;
        }
        map.put(target - num, i);
    }
    return res;
}