文章目录
454. 四数相加II
给你四个整数数组 nums1
、nums2
、nums3
和 nums4
,数组长度都是 n
,请你计算有多少个元组 (i, j, k, l)
能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间,最终结果不会超过 2^31 - 1 。
示例1:
输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出:2
解释:
两个元组如下:
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
示例2:
输入:nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出:1
思路
如果将数组AB元素加和后一起算,CD元素加和后一起算,就是要找0-(c+d)
后是否能在(a+b)
的集合中找到,所以可以用哈希法。因为最后要求四元组的个数,所以要找到有多少个符合要求的ab元素和,如果是hashset那么只能存放key值a+b
,所以要用hashmap
因此可以定义一个hashmap,遍历A、B数组,key放a和b两数之和,value 放a和b两数之和出现的次数。遍历C、D数组,如果key值 0-(c+d) 在map中出现,那么累加统计value
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
// 定义存储ab之和的map
HashMap<Integer, Integer> hm = new HashMap<>();
// 定义count计算出现和为0的元组的个数
int count = 0;
// 遍历nums1和nums2
for (int i : nums1) {
for (int j : nums2) {
// 先定义a+b的和
int temp = i + j;
// 如果需要对value进行叠加,调用put()方法之前用containsKey()方法判断是否有重复的键值,
// 如果有,则用get()方法获取原有的value,再加上新加入的value即可。
if (!hm.containsKey(temp)) {
hm.put(temp, 1);
}else {
hm.put(temp, hm.get(temp) + 1);
}
}
}
// 遍历nums3和nums4
for (int i : nums3) {
for (int j : nums4) {
// 如果map里存在0-(a+b)
if (hm.containsKey(0 - (i + j))) {
count += hm.get(0 - (i + j));
}
}
}
return count;
}
383. 赎金信
给你两个字符串:ransomNote
和 magazine
,判断 ransomNote
能不能由 magazine
里面的字符构成。
如果可以,返回 true
;否则返回 false
。
magazine
中的每个字符只能在 ransomNote
中使用一次。
示例1:
输入:ransomNote = "a", magazine = "b"
输出:false
示例2:
输入:ransomNote = "aa", magazine = "ab"
输出:false
示例3:
输入:ransomNote = "aa", magazine = "aab"
输出:true
思路
这道题可以理解为判断第一个字符串ransom能不能由第二个字符串magazines里面的字符构成,所以应该把magzines里的词放进哈希表,又因为字符串magazines里出现的每一个字母都只能取一次,所以因此要用hashmap存key字符,value出现的次数
难点
其实在本题的情况下,使用map的空间消耗要比数组大一些的,因为map要维护红黑树或者哈希表,而且还要做哈希函数,是费时的!数据量大的话就能体现出来差别了。 所以数组更加简单直接有效!
所以在这用两个方法完成该题
数组法
public boolean canConstruct(String ransomNote, String magazine){
// 定义magzine数组
int[] record = new int[26];
// 遍历字符串magzine
for (int i = 0; i < magazine.length(); i++) {
int index = magazine.charAt(i) - 'a';
record[index]++;
}
for (int i : ransomNote.toCharArray()) {
record[i - 'a']--;
}
for (int i : record) {
if (i < 0) {
return false;
}
}
return true;
}
map法
public boolean canConstruct1(String ransomNote, String magazine){
// 定义存放magzine的map
HashMap<Character, Integer> map = new HashMap<>();
for (int i = 0; i < magazine.length(); i++) {
char temp = magazine.charAt(i);
if (!map.containsKey(temp)) {
map.put(temp, 1);
}else {
map.put(temp, map.get(temp) + 1);
}
}
for (int i = 0; i < ransomNote.length(); i++) {
char temp = ransomNote.charAt(i);
if (!map.containsKey(temp)) {
return false;
}else if (map.get(temp) > 0){
map.put(temp, map.get(temp) - 1);
}else {
return false;
}
}
return true;
}
15. 三数之和
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为 0
且不重复的三元组。
**注意:**答案中不可以包含重复的三元组。
示例1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
思路
一开始想到用hash法是因为:可以先求(a+b)
,然后在确定0-(a+b)
是否出现在数组里。但是题目要求不可以包含重复的三元组,把符合条件的三元组放进vector中,然后再去重,这样是非常费时的,很容易超时,就需要考虑其他的方法。
双指针法
-
i
遍历数组,固定数a -
左指针left:
left
指向i
的下一位即i+1
,代表数b -
右指针right:
right
指针指向lenth-1
,代表数c
考虑用双指针法:假设三元组为(a,b,c),先将原数组排序,先用i
定住数字a其中i
从前遍历到尾部,left
和right
指针从两头向中间遍历。
如果nums[i] + nums[left] + nums[right] > 0
就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。
如果nums[i] + nums[left] + nums[right] < 0
说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
难点1:a的剪枝
当我们把数组从小到大排序后,如果nums[i]大于0,那么i之后的数都不用考虑,不可能相加为0
难点2:a的去重
a的去重,a对应的就是nums[i],在a遍历数组的过程中,a如果重复了就应该跳过去,此时有两种判断写法
注意:我们要做的是不能有重复的三元组,但三元组内的元素是可以重复的!
- 判断 nums[i] 与 nums[i + 1]是否相同,即比较nums[i]的后面一个元素
在如图所示的元组中,当遍历到第一个-1 的时候,判断下一个也是-1,那这组数据就pass了。所以这种判断方式不行
- 判断 nums[i] 与 nums[i-1] 是否相同,即比较nums[i]的前面一个元素
当我们遍历到nums[i]时,即第一个 -1 的时候,只要前一位没有-1,那么 {-1, -1 ,2} 这组数据一样可以收录到结果集里。但是遍历到第二个-1的时候,因为和前面一个-1相等,就需要去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
因此要在i>0时做对数a进行去重
难点3:b与c的去重
b与c的去重可以写成:
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
去重逻辑应该放在找到一个三元组之后,对b和c去重,而不是放在一开始while (right > left)后,如果放在一开始,若元组{0,0,0}贼会直接让左/右指针开始移动,从而导致 right<=left 跳出循环
注:a、b、c的去重都是要先找到想要的东西后,再进行去重
难点4:数组转集合
Arrays.asList()方法的作用是将数组或一些元素转为集合,是JDK中java.util包中Arrays类的静态方法。
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
// 定义返回的List
List<List<Integer>> res = new LinkedList<>();
// i遍历数组,如果从第i个数开始就大于0了,排序后的数都比i大相加更不可能大于0了
for (int i = 0; i < nums.length; i++) {
if (nums[i] > 0) {
return res;
}
// a的去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
// 定义左右指针
int left = i + 1;
int right = nums.length - 1;
// 左右指针开始移动(-2,-2,-1,0,1)
while (left < right) {
if (nums[i] + nums[left] + nums[right] < 0) {
left++;
}
else if(nums[i] + nums[left] + nums[right] > 0) {
right--;
}
else {
// 如果和为0就加入List
res.add(Arrays.asList(nums[i], nums[left], nums[right]));
while (left < right && nums[right] == nums[right - 1]) right--;
while (left < right && nums[left] == nums[left + 1]) left++;
left++;
right--;
}
}
}
return res;
}
- 时间复杂度(O^2)
18. 四数之和
给你一个由 n
个整数组成的数组 nums
,和一个目标值 target
。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]]
(若两个四元组元素一一对应,则认为两个四元组重复):
-
0 <= a, b, c, d < n
-
a
、b
、c
和d
互不相同 -
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
示例1:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例2:
输入:nums = [2,2,2,2,2], target = 8
输出:[[2,2,2,2]]
思路
四数之和,和15.三数之和是一个思路,都是使用双指针法, 基本解法就是在15.三数之和的基础上再套一层for循环。
四数之和的双指针解法是两层for循环nums[k] + nums[i]
为确定值,依然是循环内有left和right下标作为双指针,找出nums[k] + nums[i] + nums[left] + nums[right] == target
的情况。
双指针法
-
i
遍历数组,固定数a -
j
遍历数组,固定数b -
左指针left:
left
指向j
的下一位即j+1
,代表数c -
右指针right:
right
指针指向lenth-1
,代表数d
考虑用双指针法:假设四元组为(a,b,c,d),先将原数组排序,先用i
定住数字a,然后j
固定数字b,从前遍历到尾部,left
和right
指针从两头向中间遍历c和d。
如果nums[i] + nums[j] + nums[left] + nums[right] > 0
就说明此时四数之和大了,因为数组是排序后了,所以right
指针就应该向左移动,这样才能让四数之和小一些。
如果nums[i] + nums[j] + nums[left] + nums[right] < 0
说明 此时四数之和小了,left
就向右移动,才能让四数之和大一些,直到left
与right
相遇为止。
难点1:a的剪枝
因为题中给的和是target,可能大于0,也可能小于0,因此不能简单地用nums[i]>target
来剪枝,因为当target为负数时,nums[i]及后面的数相加会变得更小(target=-5,{-1,-4,0,0})。
所以此时的剪枝只能是当元组中的四个数都为正,且第一数a就大于target,即nums[i] > 0 && nums[i] > target
难点2:a与b的去重
a的去重,a对应的就是nums[i],在a遍历数组的过程中,a如果重复了就应该跳过去,此时有两种判断写法,判断 nums[i] 与 nums[i + 1]是否相同,还是判断 nums[i] 与 nums[i-1] 是否相同。
注:我们要做的是不能有重复的三元组,但三元组内的元素是可以重复的!
如果target为0,元组为{-1,-1,0,2},那么nums[i] == nums[i + 1],遍历到-1时下一个也是-1,就被pass了
而nums[i] == nums[i - 1],当前使用 nums[i],我们判断前一位是不是一样的元素,当遍历到第一个 -1 的时候,只要前一位没有-1,那么 {-1,-1,0,2} 这组数据一样可以收录到结果集里。
b同理。
难点3:b的剪枝
同理a的剪枝,此时的剪枝只能是当元组中的第一个数a与第二个数b的和(a+b)以及后面两个数,三个都为正,且第一个数a与第二个数b的和(a+b)就大于target,即nums[i] + nums[j] > 0 && nums[i] + nums[j] > target
难点4:c与d的去重
和三数之和的b与c的去重相同
难点5:剪枝后的跳出
在三数之和中,剪枝后可以直接return,因为跳出后就是return函数,但是在四数之和的第二次剪枝中,直接return会忽略掉第一个数a后面还没遍历到的值,所以只能break第二个for重新回到第一个for,而不能直接return。
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);
// 定义返回的List
List<List<Integer>> res = new ArrayList<>();
// i遍历数a
for (int i = 0; i < nums.length; i++) {
// a的剪枝
if (nums[i] > 0 && nums[i] > target) {
return res;
}
// a的去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
// j遍历数b,因为不重复,所以j从i+1的位置开始
for (int j = i + 1; j < nums.length; j++) {
// j的剪枝
if (nums[i] + nums[j] > 0 && nums[i] + nums[j] > target){
break;
}
// b 的去重
if (j > i + 1 && nums[j] == nums[j - 1]){
continue;
}
// 定义左右指针
int left = j + 1;
int right = nums.length - 1;
while (left < right) {
// 定义和记录当前遍历到的四个数和
// int sum = nums[i] + nums[j] + nums[left] + nums[right];
long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
if (sum > target) {
right--;
}else if (sum < target) {
left++;
}else {
res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
// c的去重
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
// d的去重
while (left < right && nums[left] == nums[left + 1]){
left++;
}
left++;
right--;
}
}
}
}
return res;
- 时间复杂度O(N^3)