15 - 3Sum
JAVA
class Solution { public List<List<Integer>> threeSum(int[] nums) { Arrays.sort(nums); List<List<Integer>> ans = new ArrayList<>(); int n = nums.length; for (int i = 0; i < n - 2 && nums[i] <= 0; ++i) { if (i > 0 && nums[i] == nums[i - 1]) { continue; } int j = i + 1, k = n - 1; while (j < k) { int x = nums[i] + nums[j] + nums[k]; if (x < 0) { ++j; } else if (x > 0) { --k; } else { ans.add(List.of(nums[i], nums[j++], nums[k--])); while (j < k && nums[j] == nums[j - 1]) { ++j; } while (j < k && nums[k] == nums[k + 1]) { --k; } } } } return ans; } }
C++
class Solution { public: vector<vector<int>> threeSum(vector<int>& nums) { sort(nums.begin(), nums.end()); vector<vector<int>> ans; int n = nums.size(); for (int i = 0; i < n - 2 && nums[i] <= 0; ++i) { if (i && nums[i] == nums[i - 1]) { continue; } int j = i + 1, k = n - 1; while (j < k) { int x = nums[i] + nums[j] + nums[k]; if (x < 0) { ++j; } else if (x > 0) { --k; } else { ans.push_back({nums[i], nums[j++], nums[k--]}); while (j < k && nums[j] == nums[j - 1]) { ++j; } while (j < k && nums[k] == nums[k + 1]) { --k; } } } } return ans; } };
Comments
Post a Comment