题目意思是,这个人有很多袜子。socks(袜子的复数)
但是他每只袜子长度都不一样。他每天穿一对长度差最小的袜子,穿完之后第二天就换,也就是说,每一对匹配出来的袜子他只会穿一天。
求第K天他穿的袜子长度。
解法:直接暴力是不对的。先对两只袜子的长度差进行二分。然后再查找长度差的值。
关键在,计算这个长度差范围之内的袜子匹配种数。应为对袜子排序过。所以可以看出,中间一大段都是重复的计算次数的,只需要计算出第一次。之后的都只需要判断新出的匹配情况就好了。
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
int num[100002];
int n;
ll k;
ll cal(int mid)
{
ll cnt = 0;
ll tol = 0;
int i=0,j=1;
while(i<n)
{
if(j<n&&num[j]-num[i]<=mid)
j++,cnt++;
else
{
tol+=cnt;
cnt--; //I++所以种数(也可以看成长度)减1;
i++;
}
}
return tol;
}
int main()
{
while(cin >> n >> k)
{
if(n==0&&k==0)break;
for(int i=0;i<n;i++)
cin >> num[i];
sort(num,num+n);
ll l = 1,r = 1000000009;
while(l<r)
{
ll mid = (l+r)>>1;
ll tol = cal(mid);
if(tol<k)l = mid+1;
else r = mid;
}
//cout<<r<<"///"<<endl;
ll cll = k-cal(r-1);
int i = 0,j = 1;
while(i<n)
{
if(cll==0)break;
if(j<n&&num[j]-num[i]<=r)
{
if(num[j]-num[i]==r)
cll--;
if(cll==0)break;
j++;
}
else
i++;
}
cout<<num[i] <<" "<<num[j]<<endl;
}
return 0;
}
I