国庆浪了一大段时间,现在得好好补补了......
由于要补的东西比较多,所以还是选择了一般难度的题
好了,言归正传,先看题
Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.
For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0. A solution set is: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]大致理解一下,就是给出一个target,然后从S中找出4个数,加起来等于这个target,最后的结果由小到大排序。
先看代码吧
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> ans;
int n = nums.size();
if(n < 4) return ans;
sort(nums.begin(),nums.end());
for(int 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(int 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;
for(int p = j + 1; p < n - 1; p++){
if(p > j + 1&&nums[p] == nums[p - 1]) continue;
if(nums[i] + nums[j] + nums[p] + nums[p + 1] > target) break;
if(nums[i] + nums[j] + nums[p] + nums[n - 1] < target) continue;
for(int q = p + 1; q < n; q++){
if(q > p + 1&&nums[q] == nums[q - 1]) continue;
int sum = nums[i] + nums[j] + nums[p] + nums[q];
if(target == sum){
ans.push_back(vector<int>{ nums[i], nums[j], nums[p], nums[q]});
}
}
}
}
}
return ans;
}
};
大致想法就是,因为输出要由小到大嘛,那就干脆直接先排个序,再来分析。
我们先选第一个数,这个数是从第一位从小到大一个一个选的,第二个数则要从第一个数的下一位开始一个一个选,以此类推。然后4个循环,把四个数加起来,看等不等于target就行了。但是,结果并没有那么简单,系统直接给了我个超时,真棒......
所以,得优化一下,免去一些不必要的操作。
第一,在当前数的循环中,把前面的数、当前数和当前数的后几位(凑够4位)加起来,如果结果比target大,那么直接跳出当前数的循环。好像有点难懂,举个例子吧,比如说现在在第二层循环,即当前数是第二个数,就把第一个数,第二个数和第二个数的后两位数加起来,看结果是不是比target大。因为这个结果是取当前数的情况下的最小和了,如果最小和都比target大,那么当前数和之后的数就都没有意义了,直接break跳出当前数循环。
第二,在当前数的循环中,把前面的数、当前数和最后的几位(凑够4位)加起来,如果结果比target小,那么直接在当前数的循环实行continue。举个例子吧,比如说现在在第二层循环,即当前数是第二个数,就把第一个数,第二个数和倒数后两位数加起来,看结果是不是比target小。因为这个结果是取当前数的情况下的最大和了,如果最大和都比target小,那么当前数就没有意义了,需要更大的当前数,直接continue。
tip:
排完序之后,数值相同的数会黏在一起,当遇到当前数字与前一个相同时,就要直接跳过,不然会得出两个相同的结果。