编程题：不重复3SUM

题目描述

给定排序数组arr和整数k，不重复打印arr中所有相加和为k的不降序三元组
例如, arr = [-8, -4, -3, 0, 1, 2, 4, 5, 8, 9], k = 10，打印结果为：
-4 5 9
-3 4 9
-3 5 8
0 1 9
0 2 8
1 4 5

[要求]

时间复杂度为O(n^2)
空间复杂度为O(1)

输入描述:
第一行有两个整数n, k
接下来一行有n个整数表示数组内的元素
输出描述:
输出若干行，每行三个整数表示答案
按三元组从小到大的顺序输出(三元组大小比较方式为每个依次比较三元组内每个数)
示例1
输入
10 10
-8 -4 -3 0 1 2 4 5 8 9
输出
-4 5 9
-3 4 9
-3 5 8
0 1 9
0 2 8
1 4 5

CODE

//当做固定一个位置i，去找i+1到j的2sum
import java.util.*;
public class Main{
    
    public static void  main(String[] args){
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        if(N<3){
            return;
        }
        int K = scanner.nextInt();
        int[] arr  =  new int[N];
        for(int i=0;i<N;i++){
            arr[i] = scanner.nextInt();
        }
        if(arr == null){
            return;
        }
        findSum(arr,K); 
        
    }
    public static void findSum(int[] arr,int K){
        for(int i=0;i<arr.length;i++){
           //如果有重复的，要跳过~
           while(i < arr.length-1 && arr[i]==arr[i+1]){
               i++;
           }
           int key = K-arr[i];
           int z = i+1;
           int j = arr.length-1;
           HashSet<Integer> res = new HashSet<>();
           while(z<j){
               if(arr[z] + arr[j] == key){
                    if(!res.contains(arr[z])){
                        res.add(arr[z]);
                        System.out.println(arr[i]+" "+arr[z]+" "+arr[j]);
                    }else{
                        z++;
                        j--;
                    }
                }
                else if(arr[z]+arr[j] < key){
                    z++;
                }else{
                    j--;
                } 
           }
        } 
    }
}

KEYPOINT

• 循环遍历，当做每次固定一个位置i，去找i+1到j的2sum
• 如果有重复的，要跳过

  while(i < arr.length-1 && arr[i]==arr[i+1]){
              i++;
          }