標簽：數組、中等難度
問題描述
Given an array nums of n integers and an integer target, are there elements a, b, c, and d in nums such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.
Note:
The solution set must not contain duplicate quadruplets.
給定整數數組nums和目標值target,判斷數組中是否存在元素a,b,c和d使得a+b+c+d=target? 給定目標值target尋找所有的四元組，使得其和為target。
注意: 解集合中不包含重復的四元組。
示例：
給定數組nums = [1, 0, -1, 0, -2, 2], 和目標值target = 0.
解集合是：
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
解決方法
方法1：暴力破解法
任選四個元素，判斷其值是否為target。
但這里存在一個問題，如果兩個解相同，則僅保存一個解，因為題目中要求解集合中不包含重復的四元組。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
vector<int> res;
for(int i = 0; i < nums.size() - 3; i++) {
for(int j = i + 1; j < nums.size() - 2; j++) {
for(int k = j + 1; j < nums.size() - 1; k++) {
for(int m = k + 1; k < nums.size(); m++) {
if (nums[i] + nums[j] + nums[k] + nums[m] == target) {
bool found = false;
res.clear();
res.push_back(i);
res.push_back(j);
res.push_back(k);
res.push_back(m);
for(int n = 0; n < result.size(); n++) {
if (result[i] == res) {
found = true;
break;
}
}
if (found == false)
result.push_back(res);
}
}
}
}
}
return result;
}
};
算法分析：
- 程序中包含5層for循環，時間復雜度接近O(n5)。
- 運行時間：Time Limit Exceeded
方法二：排序法
首先對數組進行排序，接著任選兩個數i和j。然后，從兩端遍歷數組，尋找其和值等于target的端點k和m。由于數組有序，假設A[i]+A[j]+A[k]+A[m]小于target，則將k右移；若其和大于target，則將m左移。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
if(nums.size() < 4)
return result;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size() - 3; i++) {
for(int j = i + 1; j < nums.size() - 2; j++) {
int k = j + 1;
int m = nums.size() - 1;
while (k < m) {
if (nums[i] + nums[j] + nums[k] + nums[m] == target) {
bool found = false;
vector<int> res;
res.push_back(nums[i]);
res.push_back(nums[j]);
res.push_back(nums[k]);
res.push_back(nums[m]);
for(int n = 0; n < result.size(); n++) {
int equal = 0;
for(int p = 0; p < res.size(); p++) {
if(result[n][p] == res[p]) {
equal++;
} else {
break;
}
}
if (equal == 4)found = true;
}
if (found == false) {
result.push_back(res);
}
}
if(nums[i] + nums[j] + nums[k] + nums[m] < target)
k++;
else
m--;
}
}
}
return result;
}
};
算法分析
- 運行時間：72ms
方法三：
對方法二做了一些優化，避免重復元素。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
if(nums.size() < 4)
return result;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size() - 3; i++) {
if(i > 0 && nums[i] == nums[i - 1])
continue;
for(int j = i + 1; j < nums.size() - 2; j++) {
if (j > i + 1 && nums[j] == nums[j - 1])
continue;
int k = j + 1;
int m = nums.size() - 1;
while (k < m) {
if (nums[i] + nums[j] + nums[k] + nums[m] == target) {
vector<int> res;
res.push_back(nums[i]);
res.push_back(nums[j]);
res.push_back(nums[k]);
res.push_back(nums[m]);
result.push_back(res);
while (k < m && nums[k++] == nums[k]) {}
while (k < m && nums[m--] == nums[m]) {}
}
else if(nums[i] + nums[j] + nums[k] + nums[m] < target) {
while (k < m && nums[k++] == nums[k]) {}
}
else
while (k < m && nums[m--] == nums[m]) {};
}
}
}
return result;
}
};
算法分析
- 時間復雜度約為O(n3)
- 運行時間：52ms
