数组中和等于K的数对（ 51Nod - 1001 ）

题目链接： H - 数组中和等于K的数对

来源： 51Nod - 1001 

Problem Description

给出一个整数K和一个无序数组A，A的元素为N个互不相同的整数，找出数组A中所有和等于K的数对。例如K = 8，数组A：{-1,6,5,3,4,2,9,0,8}，所有和等于8的数对包括(-1,9)，(0,8)，(2,6)，(3,5)。

Input

第1行：用空格隔开的2个数，K N，N为A数组的长度。(2 <= N <= 50000，-10^9 <= K <= 10^9)
第2 - N + 1行：A数组的N个元素。（-10^9 <= Aii <= 10^9)

Output

第1 - M行：每行2个数，要求较小的数在前面，并且这M个数对按照较小的数升序排列。 
如果不存在任何一组解则输出：No Solution。

Sample Input

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

Sample Output

-1 9
0 8
2 6
3 5

Hint：先将序列从小到大排序，然后依次遍历，注意冒泡，快排会超时。我这里给出两种做法。

这是我自己的代码：（感觉优化很差，有多余的东西，其实可以直接用sort函数，当时没想到）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

int main()
{
    long long n, k, i, j, x, a[50050];
    priority_queue <long long> q;  // 这里建立了优先队列，不过忘记了怎么建立从小到大的了
    // priority_queue <long log, vector <long long>, greater <long long> > q;
    scanf("%I64d %I64d", &k, &n);
    for(i = 0; i < n; i++)
    {
        scanf("%I64d", &x);
        q.push(-x);  //所以这里存的是负值
    }
    for(i = 0; i < n; i++)
    {
        a[i] = -q.top(); // 然后负值输出
        q.pop();
    }
    // 其实可以直接存然后用sort函数
    int flag = 0;
    for(i = 0; i < n; i++)
    {
        for(j = i + 1; j < n; j++)
        {
            if(a[i] + a[j] == k)
            {
                flag = 1;
                printf("%I64d %I64d\n", a[i], a[j]); // 最后这里进行循环判断（1s刚好，有点险）
            }
        }
    }
    if(flag == 0) printf("No Solution\n");
    return 0;
}

下面是某个大佬的代码，我自己手敲了一遍，然后加了点注释（有写的不对的地方欢迎大家下方评论留言）：

#include <bits/stdc++.h>

using namespace std;

const int N = 5e4 + 100;
map<int, bool>flag;
int s[N];
int main()
{
    ios::sync_with_stdio(0);
    int n, k, i;
    cin>> k >> n;
    for(i = 1; i <= n; i++)
    {
        cin >> s[i];
        flag[s[i]] = 1; // 标记出现的数为1
    }
    sort(s + 1, s + n + 1); // 排序
    int mark = 0;
    for(i = 1; i <= n; i++)
    {
        if(k - s[i] <= s[i]) break;  // 如果当前的数比k还要大就break掉
        if(flag[k - s[i]])  // 判断k - s[i]是否在序列中，在就输出
        {
            mark = 1; // 标记有解决方案
            cout << s[i] << ' ' << k - s[i] << endl;
        }
    }
    if(mark == 0) cout << "No Solution" << endl;
    return 0;
}