結果

問題 No.1597 Matrix Sort
ユーザー noriocnorioc
提出日時 2024-06-13 09:46:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 635 ms / 1,500 ms
コード長 914 bytes
コンパイル時間 527 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 373,360 KB
最終ジャッジ日時 2024-06-13 09:47:15
合計ジャッジ時間 13,395 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,352 KB
testcase_01 AC 41 ms
51,968 KB
testcase_02 AC 39 ms
52,480 KB
testcase_03 AC 573 ms
372,348 KB
testcase_04 AC 582 ms
372,468 KB
testcase_05 AC 601 ms
372,464 KB
testcase_06 AC 635 ms
372,756 KB
testcase_07 AC 536 ms
373,360 KB
testcase_08 AC 563 ms
372,812 KB
testcase_09 AC 532 ms
372,952 KB
testcase_10 AC 460 ms
371,188 KB
testcase_11 AC 500 ms
372,828 KB
testcase_12 AC 108 ms
102,668 KB
testcase_13 AC 206 ms
160,888 KB
testcase_14 AC 581 ms
373,096 KB
testcase_15 AC 119 ms
102,404 KB
testcase_16 AC 174 ms
161,120 KB
testcase_17 AC 507 ms
373,104 KB
testcase_18 AC 589 ms
372,972 KB
testcase_19 AC 488 ms
372,464 KB
testcase_20 AC 483 ms
372,780 KB
testcase_21 AC 516 ms
371,076 KB
testcase_22 AC 473 ms
371,160 KB
testcase_23 AC 502 ms
370,776 KB
testcase_24 AC 87 ms
100,808 KB
testcase_25 AC 85 ms
101,252 KB
testcase_26 AC 37 ms
52,000 KB
testcase_27 AC 38 ms
51,868 KB
testcase_28 AC 37 ms
53,240 KB
testcase_29 AC 393 ms
360,588 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import accumulate


def accum(a: list):
    acc = list(accumulate(a))
    return lambda l, r: acc[r] - (acc[l-1] if l > 0 else 0)


N, K, P = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

bs = [0] * (P+1)
for b in B:
    bs[b] += 1

acc_bs = accum(bs)


# A_i + B_j <= x を満たす (i, j) の個数を返す
def count(x: int) -> int:
    res = 0
    for a in A:  # A_i を固定
        # a + B_j <= x を満たす j の範囲
        if 0 <= x-a:
            res += acc_bs(0, x-a)
        # a + B_j <= P + x を満たす j の範囲
        res += acc_bs(P-a, min(P, P-a+x))

    return res


lo = 0
hi = P
ans = hi
while lo <= hi:
    m = (lo + hi) // 2
    cnt = count(m)  # A_i + B_j <= m (mod P) を満たす個数は K 個以上か？
    if cnt >= K:
        ans = min(ans, m)
        hi = m - 1
    else:
        lo = m + 1

print(ans)
0