问题:给出一个序列集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;
}