C.Find Pair（codeforces）

Find Pair

问题：给出一个序列集S，对于<x, y> ，x 在S中， y也在S中，按字典序排列，求第k个<x, y>。

例子：1 2 3（n = 3）

<1, 1>	<1, 2>	<1, 3>
<2, 1>	<2, 2>	<2, 3>
<3, 1>	<3, 2>	<3, 3>

我们取第k个<x, y>。

我刚开始直接暴力（Tle了）

再想到下面的。

按顺序排列，每个数字组成n个 <x, y>。我们可以发现（k-1）/ n 为第一个数字x（刚开始想的是k/n，发现不行）。再有点野心试试可以用 k 和 n 找到第二个数字y，就有了（k-1）% n 是y。

我又兴高采烈的去交，还交了不止一发（wa了）

又有想到了有重复元素的情况，再来个例子：1 1 2 3（n = 4）.

 

<1, 1>	<1, 1>	<1, 2>	<1, 3>
<1, 1>	<1, 1>	<1, 2>	<1, 3>
<2, 1>	<2, 1>	<2, 2>	<2, 3>
<3, 1>	<3, 1>	<3, 2>	<3, 3>

然后就去重，循环，各种乱七八糟的（Tle了）

再来，找每个元素的个数，去重，确定<x, y>的第一个元素x，除去第一个元素x之前的元素组成的<x, y>（k-前面元素个数 * n）。再然后就遍历确定第二个元素（k - 用第一个元素个数 * 各个元素的个数，直到 k <= 0）

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
ll s[100010], t[100010];
char u[20], v[20];
map<int, int> M;
int main() {
	ll n, m, w, r;
	scanf("%I64d%I64d", &n, &m);
	for(int i=0; i<n; i++) {
		scanf("%I64d", &s[i]);
		M[s[i]]++;
	}
	sort(s, s+n);
	ll x = s[(m-1)/n];

    //去重
	ll con = unique(s, s+n) - s;
    
    //如果相等表面没有重复元素
	if(n == con) {
		printf("%I64d %I64d", x, s[(m-1)%n]);
	}
	else {
        //打印第一个元素
		printf("%I64d ",  x);

		//确定x之前有多少个不同的元素，和x的个数
        ll k = lower_bound(s, s+con, x) - s, r = M[s[k]];

        //除去x之前元素组成的<x, y>
		for(int i=0; i<k; i++) m -= M[s[i]]*n;//

        //找第二个元素
		for(int i=0; i<con; i++) {
			if(m > 0) {
				m -= r*M[s[i]];
			}
			if(m <= 0) {
				printf("%I64d", s[i]);
				break;
			}
		}
	}
	return 0;
}