15. 3Sum

一开始是用回溯做的,结果超时,代码如下:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
       vector<vector<int>> ans;
       sort(nums.begin(),nums.end()); //排序,便于剔除重复答案
       vector<int> temp;
       find(nums,ans,temp,0,0);       
       return ans;
        
    }
    void find(vector<int>& nums,vector<vector<int>>& ans, vector<int>& temp,int begin,int sum)
    {
        if(temp.size()==3&&sum==0)
        {         
            ans.push_back(temp);
            return ;
        }
        
        for(int i = begin;i<nums.size();i++)
        {
            if(i>begin&&nums[i]==nums[i-1])
                continue;  //去重
            temp.push_back(nums[i]);
            find(nums,ans,temp,i+1,sum+nums[i]);
            temp.pop_back();
            
        }
    }
};

于是改为时间复杂度为o(n*n)的“双指针”推广法:

class Solution {
public:
   vector<vector<int>> threeSum(vector<int>& nums) {
	 vector<vector<int>> ans;
	 vector<int> temp;
	 int i=0;
       if(nums.size()<3)
           return ans;
	 sort(nums.begin(),nums.end());
	 while(i<nums.size()-2)
	 {
		 int begin=i+1,end=nums.size()-1;
		 while(begin<end)
		 {
			 int sum=nums[i]+nums[begin]+nums[end];
			 if(sum==0)
			 {
				 temp.push_back(nums[i]);
                 temp.push_back(nums[begin]);
                 temp.push_back(nums[end]);
				 ans.push_back(temp);
				 temp.clear();
				 begin++;
				 end--;
				 while(nums[begin]==nums[begin-1])
					 begin++;
				 while(nums[end]==nums[end+1])
					 end--;
			 }
			 else
			 {
				 if(sum<0)
					 begin++;
				 else
					 end--;
			 }				 
		 }		 
		 i++;
		 while(nums[i]==nums[i-1])
			 i++;		 
	 }	 
	 return ans;
 }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值