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

2024-08-22 09:04

文章标签 力扣地方两数

本文主要是介绍力扣: 两数之和梦开始的地方，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

文章目录

需求
暴力求解
优化一下暴力解法
用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;
}