一、题目描述
给定一个整数数组 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 <= 10^4
-10^9 <= nums[i] <= 10^9
-10^9 <= target <= 10^9
- 只会存在一个有效答案
二、解题思路
-
创建哈希表:首先,创建一个哈希表(HashMap),用于存储数组元素及其对应的索引。这样可以快速查找目标值与当前元素差值对应的另一个元素的索引。
-
遍历数组:然后,遍历数组
nums
中的每个元素。对于数组中的每个元素nums[i]
,计算目标值target
与当前元素的差值complement
。 -
查找补数:在哈希表中查找差值
complement
是否存在。如果存在,说明找到了两个数的和等于目标值,并且这两个数的索引分别存储在哈希表中和当前循环的索引i
。 -
返回结果:如果找到了补数,返回这两个数的索引。如果没有找到,继续遍历数组。
-
处理无解情况:如果遍历完整个数组都没有找到满足条件的两个数,返回一个空数组或者抛出异常,表示没有找到符合条件的两个数。
三、具体代码
import java.util.HashMap;
import java.util.Map;
public class Solution {
public int[] twoSum(int[] nums, int target) {
// 创建一个哈希表(HashMap)来存储数组元素及其对应的索引
Map<Integer, Integer> indexMap = new HashMap<>();
// 遍历数组,对于每个元素 nums[i]:
for (int i = 0; i < nums.length; i++) {
// 计算目标值与当前元素的差值
int complement = target - nums[i];
// 检查差值是否已经在哈希表中
if (indexMap.containsKey(complement)) {
// 如果存在,说明找到了两个数,返回它们的索引
return new int[]{indexMap.get(complement), i};
}
// 将当前元素及其索引存入哈希表
indexMap.put(nums[i], i);
}
// 如果遍历完数组都没有找到,返回空数组(或者抛出异常,根据题目要求)
return new int[0];
}
public static void main(String[] args) {
Solution solution = new Solution();
int[] nums = {2, 7, 11, 15};
int target = 9;
int[] indices = solution.twoSum(nums, target);
if (indices.length > 0) {
System.out.println("Indices of the two numbers: " + indices[0] + " and " + indices[1]);
} else {
System.out.println("No two sum solution");
}
}
}
四、时间复杂度和空间复杂度
-
时间复杂度:这个算法的时间复杂度是O(n),其中n是数组的长度。这是因为我们只需要遍历数组一次,并且哈希表的插入和查找操作的平均时间复杂度都是O(1)。
-
空间复杂度:空间复杂度同样是O(n),因为我们需要额外的空间来存储哈希表,它最多需要存储n个键值对。
五、总结知识点
-
哈希表(HashMap):哈希表是一种基于哈希函数的数据结构,它可以在平均情况下实现 O(1) 时间复杂度的插入和查找操作。在这个例子中,我们使用哈希表来存储数组中的元素及其对应的索引,以便快速查找目标值的补数。
-
键值对(Key-Value Pair):在哈希表中,每个元素的值(
nums[i]
)作为键,其在数组中的索引(i
)作为值。这样,我们可以通过目标值减去当前元素的值来查找补数的索引。 -
遍历数组:我们使用一个for循环遍历数组,对于每个元素,计算其与目标值的差值,并检查这个差值是否在哈希表中。如果存在,说明找到了满足条件的两个数,返回它们的索引。
-
异常处理:在实际应用中,如果题目要求在没有找到满足条件的两个数时返回空数组,我们可以直接返回。如果题目要求抛出异常,我们可以在循环结束后抛出一个
IllegalArgumentException
。
以上就是解决这个问题的详细步骤,希望能够为各位提供启发和帮助。