結果

問題 No.1597 Matrix Sort
ユーザー w0mbat
提出日時 2021-07-10 09:39:04
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 340 ms / 1,500 ms
コード長 740 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 82,520 KB
実行使用メモリ 169,496 KB
最終ジャッジ日時 2024-07-02 01:46:04
合計ジャッジ時間 6,923 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
def main():
    input = sys.stdin.readline
    N, K, P = map(int, input().split())
    *A, = map(int, input().split())
    *B, = map(int, input().split())
    A.sort()
    B.sort()
    C = [0] * (P)
    for b in B:
        C[b] += 1
    for i in range(1, P):
        C[i] += C[i - 1]

    # x以下の要素の数
    def f(x):
        res = 0
        for a in A:
            # Ai+Bj <= x
            if x - a >= 0:
                res += C[x - a]
            # P <= Ai+Bi <= P+x
            res += C[min(P - 1, P + x - a)] - C[P - a - 1]
        return res

    l = -1
    r = P - 1
    while r - l > 1:
        m = (l + r) // 2
        if f(m) < K: l = m
        else: r = m
    print(r)

if __name__ == '__main__':
    main()
0