#676821 (C++17) No.1597 Matrix Sort

• 問題: No.1597 Matrix Sort
• ユーザー: kenta255
• 提出日時: 2021-07-10 02:38:56
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 367 ms / 1,500 ms
• コード長: 1,244 bytes
• コンパイル時間: 2,256 ms
• コンパイル使用メモリ: 206,056 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-01 21:42:38
• 合計ジャッジ時間: 9,094 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	355 ms; 5,376 KB
testcase_04	AC	367 ms; 5,376 KB
testcase_05	AC	361 ms; 5,376 KB
testcase_06	AC	363 ms; 5,376 KB
testcase_07	AC	363 ms; 5,376 KB
testcase_08	AC	290 ms; 5,376 KB
testcase_09	AC	300 ms; 5,376 KB
testcase_10	AC	203 ms; 5,376 KB
testcase_11	AC	173 ms; 5,376 KB
testcase_12	AC	162 ms; 5,376 KB
testcase_13	AC	290 ms; 5,376 KB
testcase_14	AC	337 ms; 5,376 KB
testcase_15	AC	128 ms; 5,376 KB
testcase_16	AC	190 ms; 5,376 KB
testcase_17	AC	215 ms; 5,376 KB
testcase_18	AC	350 ms; 5,376 KB
testcase_19	AC	255 ms; 5,376 KB
testcase_20	AC	214 ms; 5,376 KB
testcase_21	AC	207 ms; 5,376 KB
testcase_22	AC	198 ms; 5,376 KB
testcase_23	AC	192 ms; 5,376 KB
testcase_24	AC	49 ms; 5,376 KB
testcase_25	AC	47 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	2 ms; 5,376 KB
testcase_28	AC	2 ms; 5,376 KB
testcase_29	AC	2 ms; 5,376 KB

ソースコード

#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以下の要素をかぞえる
	// 二分探索

	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 ok=-1; // K番目以下
	ll ng=P+1; // K番目より大きい

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

	cout << ng << endl;
}