#676548 (C++14) No.1597 Matrix Sort

• 問題: No.1597 Matrix Sort
• ユーザー: seven_three
• 提出日時: 2021-07-09 23:40:16
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 405 ms / 1,500 ms
• コード長: 1,177 bytes
• コンパイル時間: 948 ms
• コンパイル使用メモリ: 98,796 KB
• 実行使用メモリ: 5,708 KB
• 最終ジャッジ日時: 2024-07-01 18:31:13
• 合計ジャッジ時間: 8,052 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
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	372 ms; 5,540 KB
testcase_04	AC	389 ms; 5,576 KB
testcase_05	AC	405 ms; 5,580 KB
testcase_06	AC	394 ms; 5,576 KB
testcase_07	AC	404 ms; 5,620 KB
testcase_08	AC	299 ms; 5,460 KB
testcase_09	AC	313 ms; 5,576 KB
testcase_10	AC	202 ms; 5,580 KB
testcase_11	AC	173 ms; 5,452 KB
testcase_12	AC	164 ms; 5,448 KB
testcase_13	AC	302 ms; 5,452 KB
testcase_14	AC	341 ms; 5,708 KB
testcase_15	AC	127 ms; 5,456 KB
testcase_16	AC	190 ms; 5,456 KB
testcase_17	AC	215 ms; 5,576 KB
testcase_18	AC	371 ms; 5,452 KB
testcase_19	AC	261 ms; 5,576 KB
testcase_20	AC	210 ms; 5,580 KB
testcase_21	AC	200 ms; 5,580 KB
testcase_22	AC	201 ms; 5,452 KB
testcase_23	AC	190 ms; 5,452 KB
testcase_24	AC	50 ms; 5,452 KB
testcase_25	AC	47 ms; 5,576 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 <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cmath>
#include <stdio.h>
#include <queue>
#include <deque>
#include <cstdio>
#include <set>
#include <map>
#include <bitset>
#include <stack>
#include <cctype>
using namespace std;
long long n, k, p;
vector<long long> a, b;
bool solve(long long x) {
	long long sum = 0;
	for (int i = 0; i < n; i++) {
		long long l1 = x - a[i];
		auto itr = upper_bound(b.begin(), b.end(), l1);
		sum += int(itr - b.begin());
		long long l2 = p - a[i];
		long long r1 = p + x - a[i];
		auto itr1 = lower_bound(b.begin(), b.end(), l2);
		auto itr2 = upper_bound(b.begin(), b.end(), r1);
		sum += int(itr2 - itr1);
	}
	if (sum < k) {
		return true;
	}
	else {
		return false;
	}
}
int main() {
	cin >> n >> k >> p;
	long long a1, b1;
	for (int i = 0; i < n; i++) {
		cin >> a1;
		a.emplace_back(a1);
	}
	for (int i = 0; i < n; i++) {
		cin >> b1;
		b.emplace_back(b1);
	}
	sort(a.begin(), a.end());
	sort(b.begin(), b.end());
	long long l = -1, r = p;
	while (r - l > 1) {
		long long mid = (l + r) / 2;
		if (solve(mid)) {
			l = mid;
		}
		else {
			r = mid;
		}
	}
	cout << r << endl;
}