結果

問題 No.1597 Matrix Sort
ユーザー NS
提出日時 2021-07-10 14:34:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 356 ms / 1,500 ms
コード長 847 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 181,976 KB
最終ジャッジ日時 2024-07-02 02:24:50
合計ジャッジ時間 7,570 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,k,p=map(int, input().split())
al=list(map(int, input().split()))
bl=list(map(int, input().split()))
al.sort()
bl.sort()

bcl=[0]*(p+1)
for b in bl:
    bcl[b]+=1
for i in range(p):
    bcl[i+1]+=bcl[i]

# print(bcl)

def check_cnt(val):
    res=0
    for a in al:
        # cnt1 = bisect_right(bl, val-a)
        # # cnt2 = n - bisect_left(bl, p-a)
        # cnt2 = n - bisect_right(bl, p-a-1)
        # cnt3 = n - bisect_right(bl, val+p-a)
        # print(val-a, p-a-1)
        if val-a<0:
            cnt1=0
        else:
            cnt1=bcl[max(val-a,0)]
        cnt2=bcl[min(p,val+p-a)]-bcl[p-a-1]
        res+=(cnt1+cnt2)
    return res

ok, ng = 0, p
ok, ng = p-1, -1
while abs(ok-ng) > 1:
    mid = (ok+ng)//2
    res = True
    # ...
    cnt=check_cnt(mid)
    # print('---',mid,cnt)
    if cnt>=k: ok = mid
    else: ng = mid
print(ok)
0