2020字节跳动的社招算法 三数之和

引言  

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

   1：通过双循环 + 二分查找算法获取结果

public List<List<Integer>>  threeSum1(int[]  nums){
        List<List<Integer>> result=new ArrayList();
        if(null==nums || nums.length<=3){
          return result;
        }
        //先对数组排序
        Arrays.sort(nums);
        //通过双循环+二分查找获取结果
        for(int beginIndex= 0 ; beginIndex < nums.length-2 && nums[beginIndex]<=0;beginIndex++){
           if(beginIndex>0 && nums[beginIndex]==nums[beginIndex-1]) continue;//已经匹配过的元素直接跳过
            for(int endIndex=nums.length-1 ; endIndex > beginIndex+1 && nums[endIndex]>=0;endIndex--){
                if(endIndex<(nums.length-1) && nums[endIndex]==nums[endIndex+1]) continue;//已经匹配过的元素直接跳过
                int temp=-(nums[beginIndex]+nums[endIndex]);
                int searchIndex=Arrays.binarySearch(nums,beginIndex+1,endIndex,temp);
                if(searchIndex>0){
                    result.add(new ArrayList<Integer>(Arrays.asList(nums[beginIndex],nums[endIndex],nums[searchIndex])));
                }
            }
        }
        return result;
    }

package com.svse.test;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class SuanFaSum {

    public static void main(String[] args) {
         int[]  nums = {-1, 0, 1, 2, -1, -4, -2};
         SuanFaSum sum=new SuanFaSum();
         long start=System.currentTimeMillis();
         List<List<Integer>> allOptions=sum.threeSum1(nums);
         System.out.printf("数组%s 中三数之和为0的组合有：\n",Arrays.toString(nums));
         for(List<Integer> option:allOptions){
             System.out.printf("%s \n",option); 
         }
         long end=System.currentTimeMillis();
         System.out.println("用时："+(end-start)+"毫秒");

    }
}

 

 

 2：双索引搜索算法

public List<List<Integer>>  threeSum2(int[]  nums) {
        List<List<Integer>> result = new ArrayList();
        if (null == nums || nums.length <= 3) {
            return result;
        }
        //先对数组排序
        Arrays.sort(nums);
        //双索引搜索算法
        for(int i=0;i<nums.length-2 && nums[i]<=0;i++){
            if(i>0 && nums[i]==nums[i-1]) continue;//已经匹配过的元素直接跳过
            int beginIndex=i+1;
            int endIndex=nums.length-1;
            while (beginIndex<endIndex){
                if(beginIndex>(i+1) && nums[beginIndex]==nums[beginIndex-1]){
                    beginIndex++;
                }
                if(endIndex <(nums.length-1) && nums[endIndex]==nums[endIndex+1]){
                    endIndex--;
                }
                int temp=nums[i]+nums[beginIndex]+nums[endIndex];
                //根据当前三个数的和，判断是起始索引向后移动还是结束索引向前移动，如果为0，则两个索引同时移动
                if(temp>0){
                    endIndex--;
                }else if(temp<0){
                    beginIndex++;
                }else{
                    result.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[beginIndex],nums[endIndex])));
                    beginIndex++;
                    endIndex--;
                }
            }
        }

        return result;
    }

package com.svse.test;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class SuanFaSum {

    public static void main(String[] args) {
         int[]  nums = {-1, 0, 1, 2, -1, -4, -2};
         SuanFaSum sum=new SuanFaSum();
         long start=System.currentTimeMillis();
         List<List<Integer>> allOptions=sum.threeSum2(nums);
         System.out.printf("数组%s 中三数之和为0的组合有：\n",Arrays.toString(nums));
         for(List<Integer> option:allOptions){
             System.out.printf("%s \n",option); 
         }
         long end=System.currentTimeMillis();
         System.out.println("用时："+(end-start)+"毫秒");

    }
}