feat: 18_4sum

2025-02-20 23:35:00 +08:00 · 2025-02-20 23:35:00 +08:00 · 8a167ebab1
commit 8a167ebab1
parent a9b9675d7f
1 changed files with 36 additions and 0 deletions
--- a/18_4sum/main.cpp
+++ b/18_4sum/main.cpp
@ -0,0 +1,36 @@
+#include <bits/stdc++.h>
+using namespace std;
+
+class Solution {
+   public:
+    vector<vector<int>> fourSum(vector<int>& nums, int target) {
+        set<vector<int>> resultSet;
+
+        sort(nums.begin(), nums.end());
+
+        for (int i = 0; i < nums.size(); i++) {
+            for (int j = i + 1; j < nums.size(); j++) {
+                int l = j + 1, r = nums.size() - 1;
+
+                while (l < r) {
+                    long long sum = (long long)nums[i] + nums[j] + nums[l] + nums[r];
+                    if (sum < target) {
+                        l++;
+                    } else if (sum > target) {
+                        r--;
+                    } else {
+                        resultSet.insert({nums[i], nums[j], nums[l], nums[r]});
+                        l++;
+                    }
+                }
+            }
+        }
+
+        vector<vector<int>> result;
+        for (auto& e : resultSet) {
+            result.emplace_back(e);
+        }
+
+        return result;
+    }
+};