AI智能
改变未来

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)

[code]#include<bits/stdc++.h>#define ll long longusing 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;}

 

赞(0) 打赏
未经允许不得转载:爱站程序员基地 » C.Find Pair(codeforces)