程序员代码面试指南刷题--第八章.不重复打印排序数组中相加和为给定值的所有三元组

题目描述

给定排序数组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

解法一：三指针

import java.io.*;
public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] s1 = br.readLine().trim().split(" ");
        int len = Integer.parseInt(s1[0]);
        int k = Integer.parseInt(s1[1]);
        String[] s2 = br.readLine().trim().split(" ");
        int[] arr = new int[len];
        for(int i=0;i<len;i++){
            arr[i] = Integer.parseInt(s2[i]);
        }
        func(arr,k);
    }
    public static void func(int[] arr,int k){
        if(arr==null||arr.length<3){
            return ;
        }
        for(int i=0;i<arr.length-2;i++){
            if(i==0||arr[i]!=arr[i-1]){
                find(arr,i,i+1,arr.length-1,k-arr[i]);
            }
        }
    }
    public static void find(int[] arr,int i,int l,int r,int k){
        while(l<r){
            if(arr[l]+arr[r]<k){
                l++;
            }else if(arr[l]+arr[r]>k){
                r--;
            }else{
                if(l==i||arr[l]!=arr[l-1]){
                    System.out.println(arr[i]+" "+arr[l]+" "+arr[r]);
                }
                l++;
                r--;
            }
        }
    }
}