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

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

题目描述

给定排序数组arr和整数k，不重复打印arr中所有相加和为k的不降序二元组

例如, arr = [-8, -4, -3, 0, 1, 2, 4, 5, 8, 9], k = 10，打印结果为：

1, 9

2, 8

[要求]

时间复杂度为$O(n)$，空间复杂度为$O(1)$。

输入描述

第一行有两个整数n, k
接下来一行有n个整数表示数组内的元素

输出描述

输出若干行，每行两个整数表示答案
按二元组从小到大的顺序输出(二元组大小比较方式为每个依次比较二元组内每个数)

示例1

输入

10 10
-8 -4 -3 0 1 2 4 5 8 9

输出

1 9
2 8

备注：

$1\le N\le 10^5$

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

解法：

使用双指针算法：

1：假设 $i,j$ 分别指向序列 $arr$ 的首尾元素；

2：如果 $ar{r}_i+ar{r}_j>K$ ，说明右边指针指向的元素太大，需要往左移动，即$j$– ；

​ 如果 $ar{r}_i+ar{r}_j<K$ ，说明左边指针指向的元素太小，需要往左移动，即$i$++；

​ 否则，即找到 $ar{r}_i+ar{r}_j=K$ ，此时左指针需要向右移动，右指针需要向左移动；

​ （注意：跳过相同的元素可以避免重复元组）

代码：

#include <cstdio>

using namespace std;

const int N = 100010;
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);
    int tmp;
    for (int i = 0, j = n - 1; i < j; ) {
        if (i && a[i] == a[i - 1]) {
            ++i;
            continue;
        }
        if (j < n - 1 && a[j] == a[j + 1]) {
            --j;
            continue;
        }
        tmp = a[i] + a[j];
        if (tmp == k) {
            printf("%d %d\n", a[i], a[j]);
            ++i, --j;
        } else if (tmp > k) --j;
        else ++i;
    }
    return 0;
}