-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path3sum.cpp
68 lines (59 loc) · 1.93 KB
/
3sum.cpp
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
using namespace std;
#include <iostream>
#include <bits/stdc++.h>
// remove above lines when running in leetcode.
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
map<vector<int>,int> s;
vector<vector<int>> ans;
int len = nums.size()-1;
int value1,target=0;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++){
value1 = nums[i];
target = target - nums[i];
int left=i+1 ,right=len;
while(left<right){
if(nums[left]+nums[right]==target){
vector<int> v({value1,nums[left],nums[right]});
s[v]++;
left++;
right--;
} else if(nums[left]+nums[right]>target){
right--;
} else {
left++;
}
}
target = 0;
}
for (auto i : s) {
ans.push_back(i.first);
}
// my first approach- this was right too but was giving TLE.
// for(int i=0;i<nums.size();i++){
// value1 = nums[i];
// target = target - nums[i];
// unordered_map<int,int> umap;
// for(int j=i+1;j<nums.size();j++){
// umap[nums[j]] = j;
// }
// for(int j=i+1;j<nums.size();j++){
// if(umap.find(target - nums[j]) != umap.end() && j!=umap[target - nums[j]]){
// value2 = nums[j];
// value3 = target-nums[j];
// vector<int> v({value1,value2,value3});
// sort(v.begin(),v.end());
// // s.insert(v);
// s[v]++;
// }
// }
// target = 0;
// }
// for (auto i : s) {
// ans.push_back(i.first);
// }
return ans;
}
};