結果

問題 No.1597 Matrix Sort
コンテスト
ユーザー bokusunny
提出日時 2021-07-17 16:29:26
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 905 ms / 1,500 ms
コード長 1,081 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,766 ms
コンパイル使用メモリ 198,652 KB
最終ジャッジ日時 2025-01-23 03:01:22
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

int N, P;
long long K;
vector<int> A, B;

long long get_cnt(int x) {
  long long cnt = 0;
  for (auto a : A) {
    auto i = lower_bound(B.begin(), B.end(), P - a) - B.begin();

    // 0 <= E <= x
    int ok = -1, ng = i;

    while (abs(ok - ng) > 1) {
      int mid = (ok + ng) / 2;
      if (a + B[mid] <= x) {
        ok = mid;
      } else {
        ng = mid;
      }
    }

    // P <= E <= P+x
    int ok2 = i - 1, ng2 = N;

    while (abs(ok2 - ng2) > 1) {
      int mid2 = (ok2 + ng2) / 2;
      if (a + B[mid2] <= P + x) {
        ok2 = mid2;
      } else {
        ng2 = mid2;
      }
    }
    cnt += ok + ok2 - i + 2;
  }

  return cnt;
}

int main() {
  cin >> N >> K >> P;
  A.resize(N), B.resize(N);
  for (int i = 0; i < N; i++) cin >> A[i];
  for (int i = 0; i < N; i++) cin >> B[i];
  sort(B.begin(), B.end());

  int ng = -1, ok = P;

  while (abs(ok - ng) > 1) {
    int mid = (ok + ng) / 2;
    if (get_cnt(mid) >= K) {
      ok = mid;
    } else {
      ng = mid;
    }
  }

  cout << ok << endl;

  return 0;
}
0