結果

問題 No.507 ゲーム大会(チーム決め)
ユーザー rpy3cpp
提出日時 2017-04-23 23:12:30
言語 Python3
(3.8.2 + numpy 1.14.5 + scipy 1.1.0)
結果
AC  
実行時間 524 ms / 3,000 ms
コード長 1,014 Byte
コンパイル時間 49 ms
使用メモリ 9,836 KB
最終ジャッジ日時 2020-05-12 09:24:59

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 17 ms
5,644 KB
testcase_01 AC 16 ms
5,640 KB
testcase_02 AC 17 ms
5,640 KB
testcase_03 AC 15 ms
5,640 KB
testcase_04 AC 16 ms
5,640 KB
testcase_05 AC 17 ms
5,640 KB
testcase_06 AC 16 ms
5,640 KB
testcase_07 AC 223 ms
9,832 KB
testcase_08 AC 242 ms
9,836 KB
testcase_09 AC 415 ms
9,524 KB
testcase_10 AC 418 ms
9,528 KB
testcase_11 AC 425 ms
9,532 KB
testcase_12 AC 402 ms
9,528 KB
testcase_13 AC 524 ms
9,832 KB
testcase_14 AC 498 ms
9,824 KB
testcase_15 AC 505 ms
9,828 KB
testcase_16 AC 501 ms
9,824 KB
testcase_17 AC 17 ms
5,640 KB
testcase_18 AC 17 ms
5,640 KB
testcase_19 AC 16 ms
5,640 KB
testcase_20 AC 16 ms
5,636 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
def read_data():
    N, M = map(int, input().split())
    a0 = int(input())
    As = [int(input()) for _ in range(N - 1)]
    return N, M, a0, As


def solve(N, M, a0, As):
    As.sort(reverse=True)
    rank_best = calc_rank(a0, 0, As)
    if rank_best > M:
        return -1
    else:
        OK = 0
    rank_worst = calc_rank(a0, len(As) - 1, As)
    if rank_worst <= M:
        return As[-1]
    else:
        NG = len(As) - 1
    while OK < NG - 1:
        mid = (OK + NG) // 2
        rank = calc_rank(a0, mid, As)
        if rank <= M:
            OK = mid
        else:
            NG = mid
    return As[OK]


def calc_rank(a0, pos, As):
    score = a0 + As[pos]
    j = len(As) - 1
    rank = 1
    for i in range(len(As)):
        if i == pos: continue
        if i >= j: break
        ai = As[i]
        while (ai + As[j] <= score) or j == pos:
            j -= 1
            if i >= j: return rank
        rank += 1
        j -= 1
    return rank

N, M, a0, As = read_data()
print(solve(N, M, a0, As))
0