題目描述:
此題如果純暴力的話應該是會超時的(沒試過)考慮在暴力的基礎上優化一下,先對數組排序,然後枚舉第一個數 i,那麼剩下兩個數之和爲 j + k = 0 - i,然後設立兩個指針j和k指向頭和尾,對剩下的數組從兩頭到中間搜索答案,時間複雜度爲o(n方),代碼如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size() < 3) return res;
sort(nums.begin(),nums.end());
int j,k,target;
for(int i = 0; i < nums.size() - 2; i++){
if(i != 0){
if(nums[i] == nums[i-1]) continue;//避免重複搜索,比如 -2,-2,0,1,2。。。。。搜完第一個-2應跳過第二個-2
}
j = i + 1;
k = nums.size() - 1;
target = 0 - nums[i];
while(j < k){
if(nums[j] + nums[k] == target){
res.push_back({nums[i],nums[j],nums[k]});
while(nums[j] == nums[j+1] && nums[k] == nums[k-1]){//這裏是爲了避免形如 -2,0,0,2,2這種情況
j++;
k--;
}
j++;
k--;
}
else if(nums[j] + nums[k] < target) j++;
else k--;
}
}
return vector<vector<int>>(res.begin(),res.end());
}
};