1. 程式人生 > >51nod 1105 第K大的數

51nod 1105 第K大的數

one sum 51nod pro b+ int long sort fin

1105 第K大的數 基準時間限制:1 秒 空間限制:131072 KB 數組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
-----
李陶冶 (題目提供者) -------- 解:二分套二分;(類似於三分套三分) 技術分享
#include<cstdio>
#include<cstring>
#include<algorithm>
using std::sort;
const int N=50010;
#define ll long long 
ll a[N],b[N];
int n,k;
inline ll check(ll x)
{
    int sum=0;
    
for(int i=1;i<=n;i++) { // printf("!!!\n"); int l=1,r=n; ll ans,mid; while(l<r) { mid=(l+r)>>1; // printf("std:: 2 %d %d %lld\n",l,r,mid); ans=a[i]*b[mid]; if(ans>x) r=mid; else l=mid+1; } ans
=a[i]*b[l]; if(ans>x) sum+=(n-l+1); else { ans=a[i]*b[r]; if(ans>x) sum+=(n-r+1); } } return sum; } int main() { scanf("%d %d",&n,&k); for(int i=1;i<=n;i++) scanf("%lld %lld",&a[i],&b[i]); sort(a+1,a+1+n),sort(b+1,b+1+n); ll l=a[1]*b[1],r=a[n]*b[n]; ll ans=0,mid;0 while(r>l) { mid=(l+r)>>1; // printf("std 1:: %lld****\n",mid); ans=check(mid); if(ans>k-1) l=mid+1; else r=mid; } if(check(l)>k-1) printf("%lld\n",l); else printf("%lld\n",r); return 0; }
51nod 1105

=====================

51nod 1105 第K大的數