不重复打印排序数组中相加和为给定值的所有三元组

不重复打印排序数组中相加和为给定值的所有三元组

题目描述：

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

备注：

$1\le N\le 1000$

$-10^9\le ar{r}_i,K\le 10^9$

解法：

还是双指针算法，固定一个值，剩下两个值用双指针查找。但是此题有个坑，题干描述要求输出为严格升序的三元组，但这题实际只要三元组第一个元素和第二个元素不一样就行。。。

代码：

#include <cstdio>

using namespace std;

const int N = 1000;

int a[N];

int main(void) {
    int n, K;
    scanf("%d%d", &n, &K);
    for (int i = 0; i < n; ++i) scanf("%d", a + i);
    if (n < 3) return 0;
    if (a[0] + a[1] + a[2] > K) return 0;
    if (a[n - 3] + a[n - 2] + a[n - 1] < K) return 0;
    int tmp;
    for (int i = 0, j, k; i < n - 2; ++i) {
        if (i && a[i] == a[i - 1]) continue;
        if (a[i] + a[i + 1] + a[i + 2] > K) break;
        for (j = i + 1, k = n - 1; j < k;) {
            if (a[j] == a[j - 1]) {
                ++j;
                continue;
            }
            if (k != n - 1 && a[k] == a[k + 1]) {
                --k;
                continue;
            }
            tmp = a[i] + a[j] + a[k];
            if (tmp == K) {
                printf("%d %d %d\n", a[i], a[j], a[k]);
                ++j, --k;
            } else if (tmp > K) --k;
            else ++j;
        }
    }
    return 0;
}