牛骨文教育服务平台(让学习变的简单)
博文笔记

51nod 1001 数组中和等于K的数对

创建时间:2016-06-11 投稿人: 浏览次数:455

题目描述:

基准时间限制:1 秒 空间限制:131072 KB 分值: 5 难度:1级算法题 给出一个整数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 <= A[i] <= 10^9) 
Output
第1 - M行:每行2个数,要求较小的数在前面,并且这M个数对按照较小的数升序排列。
如果不存在任何一组解则输出:No Solution。
Input示例
8 9
-1
6
5
3
4
2
9
0
8
Output示例
-1 9
0 8
2 6
3 5


解题思路:

由于这道题是未经过排序的,所以可以用set来保存已存在的数据,插入或者删除的时间复杂度为O(logN),需要对每个点都进行这样的操作,因此总的时间复杂度为O(nlogn),

此外,如果题目中是已排好序的,则可以用首尾两个指针来确定数对。


#include <cstdio>
#include <algorithm>
#include <set>
using namespace std;

int main(){
    int k, n;
    scanf("%d%d", &k, &n);
    set<int> s, res;
    int tmp;
    for(int i = 0; i < n; i++){
        scanf("%d", &tmp);
        if(s.count(k-tmp)){
            res.insert(min(k-tmp, tmp) );
            s.erase(k-tmp);
        }
        else
            s.insert(tmp);
    }
    set<int>::iterator it = res.begin();
    if(it != res.end())
        for(; it != res.end(); it++)
            printf("%d %d
", *it, (k-*it));
    else
        printf("No Solution");

    return 0;
}


声明:该文观点仅代表作者本人,牛骨文系教育信息发布平台,牛骨文仅提供信息存储空间服务。