Advertisement
nikunjsoni

18

Apr 19th, 2021
100
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.31 KB | None | 0 0
  1. class Solution {
  2. public:
  3.     vector<vector<int>> fourSum(vector<int>& nums, int target) {
  4.         vector<vector<int>> ans;
  5.         int n = nums.size();
  6.         if(n<4)  return ans;
  7.         sort(nums.begin(), nums.end());
  8.         for(int i=0; i<n-3; i++){
  9.             if(i>0 && nums[i]==nums[i-1]) continue;
  10.             if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3] > target) break;
  11.             if(nums[i]+nums[n-3]+nums[n-2]+nums[n-1] < target) continue;
  12.             for(int j=i+1; j<n-2; j++){
  13.                 if(j>i+1 && nums[j]==nums[j-1]) continue;
  14.                 if(nums[i]+nums[j]+nums[j+1]+nums[j+2] > target) break;
  15.                 if(nums[i]+nums[j]+nums[n-2]+nums[n-1] < target) continue;
  16.                 int left=j+1, right=n-1;
  17.                 while(left<right){
  18.                     int sum=nums[left]+nums[right]+nums[i]+nums[j];
  19.                     if(sum < target) left++;
  20.                     else if(sum > target) right--;
  21.                     else{
  22.                         ans.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
  23.                         do{left++;}while(nums[left]==nums[left-1] && left<right);
  24.                         do{right--;}while(nums[right]==nums[right+1] && left<right);
  25.                     }
  26.                 }
  27.             }
  28.         }
  29.         return ans;
  30.     }
  31. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement