3Sum -- LeetCode

分享一下我老师大神的人工智能教程!零基础,通俗易懂!http://blog.csdn.net/jiangjunshow

也欢迎大家转载本篇文章。分享知识,造福人民,实现我们中华民族伟大复兴!

               

原题链接: https://oj.leetcode.com/problems/3sum/

这道题是Two Sum的扩展,brute force时间复杂度为O(n^3), 对每三个数进行比较。这道题和Two Sum有所不同,使用哈希表的解法并不是很方便,因为结果数组中元素可能重复,如果不排序对于重复的处理将会比较麻烦,因此这道题一般使用排序之后夹逼的方法,总的时间复杂度为O(n^2+nlogn)=(n^2),空间复杂度是O(n),代码如下:

public ArrayList<ArrayList<Integer>> threeSum(int[] num){    ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();    if(num==null || num.length<=2)        return res;    Arrays.sort(num);    for(int i=num.length-1;i>=2;i--)    {        if(i<num.length-1 && num[i]==num[i+1])            continue;         ArrayList<ArrayList<Integer>> curRes = twoSum(num,i-1,-num[i]);         for(int j=0;j<curRes.size();j++)         {             curRes.get(j).add(num[i]);         }         res.addAll(curRes);    }    return res;}private ArrayList<ArrayList<Integer>> twoSum(int[] num, int end,int target){    ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();    if(num==null || num.length<=1)        return res;    int l = 0;    int r = end;    while(l<r)    {        if(num[l]+num[r]==target)        {            ArrayList<Integer> item = new ArrayList<Integer>();            item.add(num[l]);            item.add(num[r]);            res.add(item);            l++;            r--;            while(l<r&&num[l]==num[l-1])                l++;            while(l<r&&num[r]==num[r+1])                r--;        }        else if(num[l]+num[r]>target)        {            r--;        }        else        {            l++;        }    }    return res;}
注意,在这里为了避免重复结果,对于已经判断过的数会skip掉,这也是排序带来的方便。 这道题考察的点其实和 Two Sum 差不多, Two Sum 是3Sum的一个subroutine, 不过更加综合一些,实现上更加全面,需要注意细节,面试中比较常见的一道题。此题更加复杂的扩展是4Sum,请参见4Sum -- LeetCode


           

给我老师的人工智能教程打call!http://blog.csdn.net/jiangjunshow

这里写图片描述

猜你喜欢

转载自blog.csdn.net/weixin_43418664/article/details/83986674