結果

問題 No.1597 Matrix Sort
ユーザー kenta255
提出日時 2021-07-10 02:52:45
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 529 ms / 1,500 ms
コード長 1,270 bytes
コンパイル時間 4,509 ms
コンパイル使用メモリ 197,860 KB
最終ジャッジ日時 2025-01-22 23:39:07
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include <atcoder/modint>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
typedef long long ll;
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)

#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define num_l(a,v) POSL(a,v) //v未満の要素数
#define num_eql(a,v) POSU(a,v) //v以下の要素数
#define num_h(a,v) (a.size() - POSU(a,v)) //vより大きい要素数
#define num_eqh(a,v) (a.size() - POSL(a,v)) //v以上の要素数


int main(){

	// 値 x が何番目かを考える
	// x以下の要素をかぞえる
	// 初めてK個以上になるxを探す
	// 二分探索

	ll N,K,P;
	cin >> N >> K >> P;

	vector<ll> A(N),B(N);
	FOR(i,0,N) cin >> A[i];
	FOR(i,0,N) cin >> B[i];
	sort(A.begin(), A.end());
	sort(B.begin(), B.end());

	ll ng=-1; // K個より少ない
	ll ok=P+8; // K個以上

	while(ok - ng > 1){
		ll x = (ok+ng) / 2;
		//x 以下の要素数をかぞえる
		ll cnt = 0;
		FOR(i,0,N){
			cnt += num_eql(B,x-A[i]);
			cnt += num_eql(B,x+P-A[i]);
			cnt -= num_l(B,P-A[i]);
		}
		if(cnt >= K){
			ok = x;
		}else{
			ng = x;
		}
	}

	cout << ok << endl;
}
0