【LeetCode】3Sum

最新推荐文章于 2022-01-27 22:44:32 发布

原创最新推荐文章于 2022-01-27 22:44:32 发布 · 737 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #sum #3sum

LeetCode 专栏收录该内容

86 篇文章

订阅专栏

本文深入解析了C++中3Sum问题的求解策略，通过将问题转化为2Sum问题并采用排序与双指针技巧，高效地找到所有符合条件的三元组。同时，提供了一个详细的代码示例，帮助读者理解和实现解决方案。

参考：

http://blog.youkuaiyun.com/xshalk/article/details/8148422

题目描述

3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:
    (-1, 0, 1)
    (-1, -1, 2)

题目分析

题目大意：给定一个数组，存在三个数a，b，c符合a+b+c=0。列出所有可能性。

输出要求，abc必须符合 a ≤ b ≤ c

思路：

这个可以转换为tow sum问题。逐个遍历数组，取出一个元素a，然后利用twosum问题找出两个相加等于b+c=-a。

对于2sum问题

需要先对数据进行排序。

把数组进行排序，然后首尾各定义一个指针head,tail，

如果target>(num[head]+num[tail]) tail--;

如果target<(num[head]+num[tail]) head++;

如果排序的复杂度为nlogn，那个3sum复杂度为n^2logn

代码示例

/*
编译环境CFree 5.0
博客地址：http://blog.youkuaiyun.com/Snowwolf_Yang
*/
#include 
    
     
#include 
     
      
using namespace std;
	
/*
vector
      
       
         > 后一个>前需要加一个空格 
*/

class Solution {
public:
    vector
        
         
           > threeSum(vector
          
            &num) { vector
           
            
              > ret; ret.clear(); sort(num.begin(),num.end()); for(int i=0; i!=num.size();i++){ if(i > 0 && num[i]==num[i-1]) //和前一个相同表示该子问题已经解决 continue; int j,k; j=i+1; k=num.size()-1; while(j
             
              i+1&&num[j]==num[j-1]){ //和前一个相同表示该子问题已经解决 j++; continue; } if(k
              
               0){ k--; }else if(sum<0){ j++; }else{ vector
               
                 tmp; tmp.push_back(num[i]); tmp.push_back(num[j]); tmp.push_back(num[k]); ret.push_back(tmp); j++; } } } return ret; } }; void test0() { int arr[] = {0,-1,2,3,-4,5,6,-7,8,9}; vector
                
                  numbers (arr, arr + sizeof(arr) / sizeof(int) ); Solution so; vector< vector
                 
                   > out = so.threeSum(numbers); int i = 0,j = 0, flag = 1; for(i = 0;i