LeetCode#18四数之和
题目链接
其实这道题本质上就是一个四层循环判断所有可能性…还以为题解有什么高深的算法呢(逃
为了避免有重复的结果出现,我们首先需要进行排序,然后如果是相同的数看作一种情况,为了降低时间复杂度,题解进行了几个剪枝:首先把所有大于的情况直接break掉,然后是在剩两个数的时候做了一个双指针,如果小于结果,就只判断左指针的右侧,如果大于结果,就只判断右指针的左侧
附代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
| class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { vector<vector<int>> quadruplets; if (nums.size() < 4) { return quadruplets; } sort(nums.begin(), nums.end()); int length = nums.size(); for (int i = 0; i < length - 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[length - 3] + nums[length - 2] + nums[length - 1] < target) { continue; } for (int j = i + 1; j < length - 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[length - 2] + nums[length - 1] < target) { continue; } int left = j + 1, right = length - 1; while (left < right) { int sum = nums[i] + nums[j] + nums[left] + nums[right]; if (sum == target) { quadruplets.push_back({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 quadruplets; } };
|