CodeForces 160CFind Pair（思路）

来源：互联网发布：mp3下载软件推荐编辑：程序博客网时间：2024/06/13 02:02

题意：给定n个数，两两之间形成一个数对，共有n*n个，求第k小的数对。

思路：首先，排个序，每一个数可以形成n个数对，这样就可以确定第k小数对的第一个数，就是a[(k-1)/n]，记为t1，那么比t1小的都不行，第k小数对只存在第一个数为t1的数对中。

#include <cstdio>#include <algorithm>#include <iostream>#include<vector>#include<cmath>#include<set>#include<cstring>#include<map>using namespace std;const int maxn = 1020;const int maxt = 100200;const int INF = 0x3f3f3f3f;const int mod = 1e9 + 7;const double pi = acos(-1.0);int n;int a[100010];map<int, int> ma;int main(){    long long k;    scanf("%d%I64d", &n, &k);    for(int i = 0; i < n; ++i){        scanf("%d", &a[i]);        ma[a[i]]++; //用map记下a[i]的个数    }    sort(a, a + n);    int p =(k - 1) / n;    p = a[p];    int kk = lower_bound(a, a + n, p) - a;    long long pp = k - (long long)kk * n;    long long y = ma[p]; // 所求数对的第一个数的个数    int t = unique(a, a + n) - a; // 去掉重复的数    long long cnt = 0;    int q;    for(int i = 0; i < t; ++i){ //从最小的开始加直到找到第k个，这里即第pp个。        long long x = ma[a[i]];        cnt += x * y;        if(cnt >= pp){q = a[i]; break;}    }    printf("%d %d\n", p, q);    return 0;}

1 0