leetcode 15. 三数之和

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

注意分析问题,i顺序遍历;j也顺序遍历和k逆序遍历,这样j和k相遇即可结束本次循环,可以大大降低复杂度;第一次做的时候选择i,j,k都是顺序遍历,理论上复杂度O(n3),怪不得超时。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> v;
        vector<int> vec(3);
        if(nums.size()<3) return v;
        sort(nums.begin(),nums.end());//先转换成有序向量
        int sum=0,m=0;
        bool flag = 1;
        for(int i=0;i<nums.size()-2;++i){
            m=0-nums[i];
            int j=i+1,k=nums.size()-1;
            while(j<k){
                sum=nums[j]+nums[k];//转换成两数之和的问题
                if(sum==m){
                   vec[0]=nums[i];
                   vec[1]=nums[j++];
                   vec[2]=nums[k--];
                   v.push_back(vec);
                while(j<k&&nums[j]==nums[j-1]) j++;//j跳过重复元素
                while(j<k&&nums[k]==nums[k+1]) k--;//k跳过重复元素
                }
                else if(sum>m) k--;//sum(两数和)较大时,k左移
                else j++;//sum较小时,j右移
            }
            while(i<nums.size()-2&&nums[i]==nums[i+1]) i++;//i跳过重复元素
        }
        return v;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值