给定一个整数数组 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]
第一种解法,暴力枚举!那么就是先在数组中找到第一个元素,然后再向后继续寻找,直到某两个元素的和为target为止,这种代码思想是相对容易的,具体实现需要两个for循环嵌套,时间复杂度是O(n²)
class Solution
{
public int[] twoSum(int[] nums, int target)
{
for(int i = 0;i < nums.length;i++)
{
for(int j = i+1;j<nums.length;j++)
{
if(nums[i] + nums[j] == target)
return new int [] {i,j};
}
}
throw new IllegalArgumentException("No two sum solution"); // 如果没有找到结果,抛出异常
}
}
public class LC01
{
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); // 调用twoSum方法
System.out.println("Indices: " + indices[0] + ", " + indices[1]); // 打印结果
}
}
第二种解法是哈希表。
首先我们来一起回顾一下哈希表的内容吧,哈希表本质上也是一个数组,要么是数组加上链表的结构,要么是数组加上二叉树的结构。一般来说长度大于等于8的时候,是二叉树结构,小于等于7的时候是链表结构。更加详细的内容给大家推荐这篇博客
https://blog.csdn.net/sinat_33921105/article/details/103344078
那么这道题目该如何用hash表来做呢?在前面用枚举法我们发现需要两个循环嵌套才能完成这道题目,如果能降低时间复杂度到O(n),那么算法的性能就有了大幅度提升了。
我们可以采用hash表中键值对的特性,将数组元素存为键,即key;数组下标存为对应的值,即value。
举一个例子
如果数组 nums
是 [2, 7, 11, 15]
,那么在遍历过程中,哈希表可能会按以下步骤更新:
- 遍历到
nums[0] = 2
,哈希表更新为{2=0}
。 - 遍历到
nums[1] = 7
,此时还没有找到与2
相加等于目标和的元素,所以继续更新哈希表为{2=0, 7=1}
。 - 遍历到
nums[2] = 11
,同样地,更新哈希表为{2=0, 7=1, 11=2}
。 - 遍历到
nums[3] = 15
,此时会检查哈希表中是否有target - 15 = 9 - 15 = -6
的键,显然没有,所以更新哈希表为{2=0, 7=1, 11=2, 15=3}
。
假设目标和 target
是 9
,那么当遍历到 nums[1] = 7
时,会计算 target - nums[1] = 9 - 7 = 2
。因为 2
已经在哈希表中作为键存在,并且对应的值是 0
(这是 2
在数组中的索引),所以方法可以立即返回 [0, 1]
,因为 nums[0] + nums[1] = 2 + 7 = 9
,这正是我们要找的答案。具体代码如下:
class Solution2 {
public int[] twoSum(int[] nums, int target)
{
Map<Integer, Integer> hashtable = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i)
{
if (hashtable.containsKey(target - nums[i]))
{
return new int[]{hashtable.get(target - nums[i]), i};
}
hashtable.put(nums[i], i);
}
return new int[0]; // 如果没有找到解,返回一个空数组
}
}
代码参考来自leetcode平台。