leetcode 15. 3Sum 暴力枚举

来源:互联网 发布:sql中select语句 编辑:程序博客网 时间:2024/05/22 00:14



链接:戳这里


题意:给出长度为n的整数序列,要求找出满足条件的所有不同三元组使得三个数和为0

两个三元组不相同意义为:集合中至少有一个数不同


思路:序列按升序排序,枚举a[k],接着在[0,k-1]选出两个数a[i],a[j] 相加使得和为-a[k],即为答案 复杂度O(n*n)

由于序列是升序排序,所以a[i]+a[j]的值是会随着i,j的移动而显得有单调性

当a[i]+a[j]<-a[k]:则i++  (当前的a[i]==a[i-1]的话,说明不需要再去比较i了 直接i++去重

当a[i]+a[j]>-a[k]:则j--  (当前的a[j]==a[j+1]的话,说明不需要再去比较j了,直接j--  去重

相等的话统计答案,接着i++,j--


代码:

class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        int n=nums.size();        sort(nums.begin(),nums.end());        vector<vector<int> > ans;        for(int k=2;k<n;k++){            if(k+1<n && nums[k]==nums[k+1]) continue;            int i=0,j=k-1;            int tmp=-nums[k];            while(i<j){                int sum=nums[i]+nums[j];                if(sum<tmp || (nums[i]==nums[i-1] && i>0) ) i++;                else if(sum>tmp || (nums[j]==nums[j+1] && j+1<k) ) j--;                else ans.push_back({nums[i++],nums[j--],nums[k]});             }        }        return ans;    }};




0 0