結果
| 問題 | 
                            No.1597 Matrix Sort
                             | 
                    
| コンテスト | |
| ユーザー | 
                             w0mbat
                         | 
                    
| 提出日時 | 2021-07-10 09:29:43 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 735 bytes | 
| コンパイル時間 | 468 ms | 
| コンパイル使用メモリ | 82,472 KB | 
| 実行使用メモリ | 169,672 KB | 
| 最終ジャッジ日時 | 2024-07-02 01:44:57 | 
| 合計ジャッジ時間 | 7,639 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge3 / judge4 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 25 WA * 2 | 
ソースコード
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 = 0
    r = P
    while r - l > 1:
        m = (l + r) // 2
        if f(m) < K: l = m
        else: r = m
    print(r)
if __name__ == '__main__':
    main()
            
            
            
        
            
w0mbat