51nod-1105-第K大的数

题目描述

数组A和数组B，里面都有n个整数。数组C共有n^2个整数，分别是A[0] * B[0],A[0] * B[1] ……A[1] * B[0],A[1] * B[1]……A[n - 1] * B[n - 1]（数组A同数组B的组合）。求数组C中第K大的数。
例如：A：1 2 3，B：2 3 4。A与B组合成的C包括2 3 4 4 6 8 6 9 12共9个数。

Input

第1行：2个数N和K，中间用空格分隔。N为数组的长度，K对应第K大的数。(2 <= N <= 50000，1 <= K <= 10^9)
第2 - N + 1行：每行2个数，分别是A[i]和B[i]。(1 <= A[i],B[i] <= 10^9)

Output

输出第K大的数

样例

input

3 2
1 2
2 3
3 4

output

9

题解

简单思路二分，但这不是一道简单的二分，这是二分套二分。
先二分答案l=a[1]*b[1];r=a[n]*b[n];
然后再在b数组里二分。
代码应该看得懂我直接贴代码quq

var n,k,i:longint;
    l,r,mid,ans:int64;
    a,b:array[0..50005] of longint;

procedure sort1(l,r:longint);
var i,j,mid,t:longint;
begin
  i:=l;j:=r;mid:=a[(l+r) div 2];
  while i<=j do
  begin
    while a[i]<mid do inc(i);
    while a[j]>mid do dec(j);
    if i<=j then
    begin
      t:=a[i];a[i]:=a[j];a[j]:=t;
      inc(i);dec(j);
    end;
  end;
  if l<j then sort1(l,j);
  if i<r then sort1(i,r);
end;

procedure sort2(l,r:longint);
var i,j,mid,t:longint;
begin
  i:=l;j:=r;mid:=b[(l+r) div 2];
  while i<=j do
  begin
    while b[i]<mid do inc(i);
    while b[j]>mid do dec(j);
    if i<=j then
    begin
      t:=b[i];b[i]:=b[j];b[j]:=t;
      inc(i);dec(j);
    end;
  end;
  if l<j then sort2(l,j);
  if i<r then sort2(i,r);
end;

function search(i:longint;mid:int64):longint;
var mm,l,r:longint;
begin
  l:=1;r:=n;search:=n;
  while l<=r do
  begin
    mm:=(l+r) div 2;
    if a[i]*b[mm]>=mid then
    begin
      if mm<search then search:=mm;
      r:=mm-1;
    end else l:=mm+1;
  end;
end;

function check(t:int64):boolean;
var ans,i:longint;
begin
  ans:=0;
  for i:=1 to n do
  begin
    if a[i]*b[n]<mid then continue;
    ans:=ans+n-search(i,mid)+1;
  end;
  if ans>=k then exit(true) else exit(false);
end;

begin
  readln(n,k);
  for i:=1 to n do readln(a[i],b[i]);
  sort1(1,n);
  sort2(1,n);
  l:=a[1]*b[1]; r:=a[n]*b[n];
  while l<=r do
  begin
    mid:=(l+r) div 2;
    if check(mid) then 
    begin 
      ans:=mid;l:=mid+1;
    end else r:=mid-1;
  end;
  writeln(ans)
end.