3Sum

来源:互联网 发布:windows管道命令 编辑:程序博客网 时间:2024/06/08 06:08

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


思路:这道题要找出不重复的合为0的三元组,做过了2Sum那道题会很容易想到借助HashMap,但是笔者尝试过,这样的方式很难解决重复的问题。暴力破解的做法是每次挑出一个元素值叫做a,然后从剩余的数组中找出和为-a的两元素,两个步骤的合在一块的时间复杂度为o(n*n^2)=o(n^3),而且也很难解决重复的问题。为了解决重复的问题,不妨先对数组排序,这样重复的元素一定是连续的比较方便处理,再来想想看有没有降低时间复杂度的方法。


现在数组已经排好序了,假如nums[i]+nums[j]=-a符合条件了。如果i,j两指针都是从前往后走的,必然nums[i]+nums[j]会一直增加再也不会符合条件,同理如果i,j两指针都是从后往前走,两者的值的和会一直减小也不会符合条件。只有当i,j两指针一个往前走,一个往后走才会有再次满足nums[i]+nums[j] = -a。所以考虑将i,j分别放到待寻找数组的头和尾避开那些不可能符合条件的情况来降低时间复杂度。


public class ThreeSum {    public List<List<Integer>> threeSum(int[] nums) {    List<List<Integer>> list = new ArrayList<List<Integer>>();        Arrays.sort(nums);        for(int i=0; i<nums.length-2; i++) {        if(i == 0 || (i>0&&nums[i] != nums[i-1])) {        int j=i+1; int k=nums.length-1; int sum = -nums[i];        while(j < k) {                   if(nums[j] + nums[k] == sum) {                   list.add(Arrays.asList(nums[i], nums[j], nums[k]));                   while(j<k && nums[j]==nums[j+1]) j++;                   while(k>j && nums[k]==nums[k-1]) k--;                   j++;k--;                   } else {                   if(nums[j] + nums[k] < sum) {                   j++;                   } else {                     k--;                   }                   }        }        }        }        return list;    }}