LeetCode 15. 三數之和 （時間超出限制後，參考他人答案）
15. 三數之和
給定一個包含 n 個整數的陣列 nums，判斷 nums 中是否存在三個元素 a，b，c ，使得 a + b + c = 0 ？找出所有滿足條件且不重複的三元組。
注意：答案中不可以包含重複的三元組。
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4]， 滿足要求的三元組集合為： [ [-1, 0, 1], [-1, -1, 2] ]
您是否在真實的面試環節中遇到過這道題目呢？
錯誤答案1（耗時太長）：
class Solution {
public: vector<vector<int>> threeSum(vector<int>& nums) { vector<int> tmp; vector<vector<int>> res; int len = nums.size(); int t,t1,t2,t3; if(3>nums.size()){return res;} for(int i=0;i<len;i++) { for(int j=i+1;j<len;j++) { for(int k=j+1;k<len;k++) { if(0==nums[i]+nums[j]+nums[k]) { t1=nums[i]; t2=nums[j]; t3=nums[k]; if(t1>t2) { t=t2; t2=t1; t1=t; } if(t2>t3) { t=t3; t3=t2; t2=t; } if(t1>t2) { t=t2; t2=t1; t1=t; } tmp.clear(); tmp.push_back(t1); tmp.push_back(t2); tmp.push_back(t3); if(0!=res.size()) { t=0; for(int n=0;n<res.size();n++) { if(tmp == res[n]) { t = 1; break; } } if(!t) { res.push_back(tmp); } } else { res.push_back(tmp); } } } } } return res; }
};
提交記錄
|
311 / 313 個通過測試用例 |
狀態：
超出時間限制 |
|
提交時間：0 分鐘之前 |
最後執行的輸入： [82597,-9243,62390,83030,-97960,-26521,-61011,83390,-38677,12333,75987,46091,83794,19355,-71037,-6242,-28801,324......
我的答案2：
class Solution { public: vector<vector<int>> threeSum(vector<int>& nums) { vector<int> tmp; vector<vector<int>> res; int len = nums.size(); int t,t1,t2,t3; vector<int>::iterator iter_i; vector<int>::iterator iter_j; vector<int>::iterator iter_k; vector<vector<int>>::iterator iter_res; iter_i = nums.begin(); for(int i=0;i<len;i++) { iter_j = nums.begin(); iter_j += i+1; for(int j=i+1;j<len;j++) { iter_k = nums.begin(); iter_k += j+1; for(int k=j+1;k<len;k++) { if(0==*iter_i+*iter_j+*iter_k) { t1=*iter_i; t2=*iter_j; t3=*iter_k; if(t1>t2) { t=t2; t2=t1; t1=t; } if(t2>t3) { t=t3; t3=t2; t2=t; } if(t1>t2) { t=t2; t2=t1; t1=t; } tmp.clear(); tmp.push_back(t1); tmp.push_back(t2); tmp.push_back(t3); if(0!=res.size()) { t=0; for(iter_res = res.begin();iter_res!=res.end();iter_res++) { if(tmp == *iter_res) { t = 1; break; } } if(!t) { res.push_back(tmp); } } else { res.push_back(tmp); } } iter_k++; } iter_j++; } iter_i++; } return res; } };
依舊是超出時間限制！！
於是乎在網上找了參考答案：
來源：https://blog.csdn.net/shinanhualiu/article/details/48913435
思路：solution: 因為三元組的元素是按照遞增的順序，因此我們首先需要對S進行排序。然後保持兩個前後索引來進行比較。
class Solution { public: /** * @param numbers : Give an array numbers of n integer * @return : Find all unique triplets in the array which gives the sum of zero. */ vector<vector<int> > threeSum(vector<int> &nums) { // write your code here vector<vector<int>> res; sort(nums.begin(), nums.end()); //對陣列進行排序 int i = 0 , last = nums.size()-1; while (i < last) { int a = nums[i], j = i + 1, k = last; while (j < k) { int b = nums[j], c = nums[k]; int sum = a + b + c; if (sum == 0) res.push_back({a,b,c}); if (sum <= 0) //注意不能含有相同的索引 while (nums[j] == b && j < k ) ++j; if (sum >= 0) while (nums[k] == c && j < k) --k; } while (nums[i] == a && i < last) ++i; } return res; } };
果然沒有好好思考出最好的方式，做了很多已然一看就和不會為0的遍歷，真是笨 = = ；