-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path3Sum
More file actions
103 lines (98 loc) · 3.26 KB
/
3Sum
File metadata and controls
103 lines (98 loc) · 3.26 KB
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
Given an integer array nums, return all the triplets [nums[i], nums[j], nums[k]] such that i != j, i != k, and j != k, and nums[i] + nums[j] + nums[k] == 0.
Notice that the solution set must not contain duplicate triplets.
Example 1:
Input: nums = [-1,0,1,2,-1,-4]
Output: [[-1,-1,2],[-1,0,1]]
Explanation:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0.
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0.
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0.
The distinct triplets are [-1,0,1] and [-1,-1,2].
Notice that the order of the output and the order of the triplets does not matter.
Example 2:
Input: nums = [0,1,1]
Output: []
Explanation: The only possible triplet does not sum up to 0.
Example 3:
Input: nums = [0,0,0]
Output: [[0,0,0]]
Explanation: The only possible triplet sums up to 0.
-----------------------------------------------------------------------------------------------------------------------------------------------------------------
//brute using 3 for loop
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
set<vector<int>>set;
int sum =0;
int n=nums.size();
for (int i=0;i<n-2;i++){
for(int j=i+1;j<n-1;j++){
for(int k=j+1;k<n;k++){
sum=nums[i]+nums[j]+nums[k];
if(sum==0 ){
vector<int>temp={nums[i],nums[j],nums[k]};
sort(temp.begin(),temp.end());
set.insert(temp);
}
}
}
}
vector<vector<int>>res(set.begin(),set.end());
return res;
}
};
-------------------------------------------------------------------------------
//better using hashmap
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
set<vector<int>> st;
int sum =0 , n=nums.size();
for(int i=0;i<n-1;i++){
set<int>hash;
for(int j=i+1;j<n;j++){
int third=-(nums[i]+nums[j]);
if(hash.find(third)!=hash.end()){
vector<int>temp={nums[i],nums[j],third};
sort(temp.begin(), temp.end());
st.insert(temp);
}
hash.insert(nums[j]);
}
}
vector<vector<int>>res(st.begin(),st.end());
return res;
}
};
------------------------------------------------------------------------------------------------------------
//optimal using 3 pointer
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>res;
int n=nums.size();
sort(nums.begin(),nums.end());
for(int i=0;i<n;i++){
if(i>0 && nums[i]==nums[i-1]) continue;
int j=i+1, k=n-1;
while(j<k){
int sum=nums[i]+nums[j]+nums[k];
if(sum<0){
j++;
}
else if (sum>0){
k--;
}
else{
vector<int>temp={nums[i],nums[j],nums[k]};
res.push_back(temp);
j++;
k--;
while(j<k && nums[j]==nums[j-1]) j++;
while(j<k && nums[k]==nums[k+1])k--;
}
}
}
return res;
}
};