与leetcode 15 三数之和类似
方法一:排序 + 双指针
最朴素的方法是使用四重循环枚举所有的四元组,然后使用哈希表进行去重操作,得到不包含重复四元组的最终答案。假设数组的长度是 n,则该方法中,枚举的时间复杂度为 O(n^4),去重操作的时间复杂度和空间复杂度也很高,因此需要换一种思路。
为了避免枚举到重复四元组,则需要保证每一重循环枚举到的元素不小于其上一重循环枚举到的元素,且在同一重循环中不能多次枚举到相同的元素。
为了实现上述要求,可以对数组进行排序,并且在循环过程中遵循以下两点:
- 每一种循环枚举到的下标必须大于上一重循环枚举到的下标;
- 同一重循环中,如果当前元素与上一个元素相同,则跳过当前元素。
使用上述方法,可以避免枚举到重复四元组,但是由于仍使用四重循环,时间复杂度仍是 O(n^4)。注意到数组已经被排序,因此可以使用双指针的方法去掉一重循环。
使用两重循环分别枚举前两个数,然后在两重循环枚举到的数之后使用双指针枚举剩下的两个数。假设两重循环枚举到的前两个数分别位于下标 i 和 j,其中 i<j
。初始时,左右指针分别指向下标 j+1 和下标 n-1。每次计算四个数的和,并进行如下操作:
-
如果和sum = target,则将枚举到的四个数加到答案中,然后将左指针右移直到遇到不同的数,将右指针左移直到遇到不同的数;
-
如果sum<target,则将左指针右移一位;
-
如果 sum>target,则将右指针左移一位。
使用双指针枚举剩下的两个数的时间复杂度是 O(n),因此总时间复杂度是 O(n^3)
具体实现时,还可以进行一些剪枝操作:
- 在确定第一个数之后,如果
nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target
,说明此时剩下的三个数无论取什么值,四数之和一定大于 target,因此退出第一重循环; - 在确定第一个数之后,如果
nums[i]+nums[n−3]+nums[n−2]+nums[n−1]<target
,说明此时剩下的三个数无论取什么值,四数之和一定小于 target,因此第一重循环直接进入下一轮,枚举 nums[i+1] - 在确定前两个数之后,如果
nums[i]+nums[j]+nums[j+1]+nums[j+2]>target
,说明此时剩下的两个数无论取什么值,四数之和一定大于 target,因此退出第二重循环; - 在确定前两个数之后,如果
nums[i]+nums[j]+nums[n−2]+nums[n−1]<target
,说明此时剩下的两个数无论取什么值,四数之和一定小于 target,因此第二重循环直接进入下一轮,枚举 nums[j+1]
var fourSum = function (nums, target) {
const quadruples = [];
if (nums.length < 4) return quadruples;
//排序
nums.sort((x, y) => x - y);
const n = nums.length; //数组下标范围为0~n-1
for (let i = 0; i < n - 3; i++) {
//在确定第一个数后的特殊情况判断
if (i > 0 && nums[i] === nums[i - 1]) {
continue;
}
if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) {
break;
}
if (nums[i] + nums[n - 3] + nums[n - 2] + nums[n - 1] < target) {
continue;
}
for (let j = i + 1; j < n - 2; j++) {
//在确定前两个数后的特殊情况判断
if (j > i + 1 && nums[j] === nums[j - 1]) {
continue;
}
if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) {
break;
}
if (nums[i] + nums[j] + nums[n - 2] + nums[n - 1] < target) {
continue;
}
//双指针枚举剩下的两个数
let left = j + 1,right = n - 1;
while (left < right) {
const sum = nums[i] + nums[j] + nums[left] + nums[right];
if (sum === target) {
quadruples.push([nums[i], nums[j], nums[left], nums[right]]);
while (left < right && nums[left] === nums[left + 1]) {
left++;
}
left++;
while (left < right && nums[right] === nums[right - 1]) {
right--;
}
right--;
}else if(sum<target){
left++;
}else{
right--;
}
}
}
}
return quadruples;
};
作者:LeetCode-Solution 链接:https://leetcode-cn.com/problems/4sum/solution/si-shu-zhi-he-by-leetcode-solution/ 来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。