【算法系列-哈希表】两数之和 (Map)
文章目录
- 【算法系列-哈希表】两数之和 (Map)
- 1. 两数之和(LeetCode 1, 梦开始的地方)
- 1.1 思路分析🎯
- 1.2 解题过程🎬
- 1.3 代码示例🌰
- 2. 四数相加II(LeetCode 454)
- 2.1 思路分析🎯
- 2.2 解题过程🎬
- 2.3 代码示例🌰
1. 两数之和(LeetCode 1, 梦开始的地方)
【题目链接】1. 两数之和 - 力扣(LeetCode)
1.1 思路分析🎯
这道题使用哈希表中的map结构来解决问题,不过开始之前我们需要理清楚为什么要使用哈希表:
什么时候用哈希表:当我们需要判断一个元素之前是否出现过 或 一个元素是否存在于一个集合中时,可以使用哈希表;
这道题中我们需要哈希表来帮我们记录已经遍历过的元素;为什么要使用map:这道题中我们返回的结果是元素在数组中的索引下标,对此可以使用map来进行映射,key存储元素,value存储索引下标
理清楚上述关系后就可以开始解题了
1.2 解题过程🎬
定义一个map,key存储元素,value存储索引下标 遍历数组:通过目标值target 减去当前数组下标元素 x1,得到数值x2,之后判断map中是否存在x2:
存在,则返回当前循环的索引值 i 以及 key = x2在map中的value值
不存在,则将当前数组下标元素x1 加入map中,继续循环
直到循环结束仍未返回,则返回空集合
1.3 代码示例🌰
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0;i < nums.length;i++) {
int x1 = nums[i];
int x2 = target - x1;
if (map.containsKey(x2)) {
return new int[]{i, map.get(x2)};
}
map.put(x1, i);
}
return new int[2];
}
}
2. 四数相加II(LeetCode 454)
【题目链接】454. 四数相加 II - 力扣(LeetCode)
2.1 思路分析🎯
这道题若用暴力解决的话,会直接**超时 O(n^4)**了,如下:
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int count = 0;
for (int i = 0;i < nums1.length;i++) {
for (int j = 0;j < nums2.length;j++) {
for (int x = 0;x < nums3.length;x++) {
for (int y = 0;y < nums4.length;y++) {
if (nums1[i] + nums2[j] + nums3[x] + nums4[y] == 0) {
count++;
}
}
}
}
}
return count;
}
}
所以使用哈希表来降低时间复杂度
2.2 解题过程🎬
定义 map 用来记录 数组1 和 数组2 遍历过的元素,key 为 两个数组元素每轮遍历元素之和,value 为两个数组元素之和出现的次数,之后分别执行下述操作:
- 先对 数组 nums1 和 数组 nums2 进行遍历,设 a 为 nums1中元素, b 为 nums2 中元素,将 a + b 添加到map中;
- 后对 数组 nums3 和 数组 nums4 进行遍历,设 c 为 nums3中元素, d 为 nums4 中元素,将 n = 0 - (c + d)添加到map中;若 n 在map中存在,则 count += map.get(n)
注: 这里之所以是 0 - (c+ d) 是因为我们需要将这个元素和拿到map中去进行匹配,而 0 - (c + d) = (a + b) ); 这里是 + map.get(n) 而不是count++ 是因为 n 可以与 a + b 进行组合,所以要加上对应的次数
直到循环结束返回count即可
2.3 代码示例🌰
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int count = 0;
Map<Integer, Integer> map = new HashMap<>();
for (int a : nums1) {
for (int b : nums2) {
map.put(x + y, map.getOrDefault(a + b, 0) + 1);
}
}
for (int c : nums3) {
for (int d : nums4) {
int n = 0 - (c + d);
if (map.containsKey(n)) {
count += map.get(n);
}
}
}
return count;
}
}
以上便是对两数之和类问题的介绍了!!后续还会继续分享其它算法系列内容,如果这些内容对大家有帮助的话请给一个三连关注吧💕( •̀ ω •́ )✧( •̀ ω •́ )✧✨