結果

問題 No.507 ゲーム大会(チーム決め)
ユーザー rpy3cpp
提出日時 2017-04-23 23:12:30
言語 Python3
(3.7.4 + numpy 1.14.5 + scipy 1.1.0)
結果
AC  
実行時間 603 ms
コード長 1,014 Byte
コンパイル時間 50 ms
使用メモリ 9,792 KB
最終ジャッジ日時 2019-09-20 01:08:40

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
0_sample1.txt AC 17 ms
6,872 KB
0_sample2.txt AC 18 ms
6,876 KB
1_test1.txt AC 18 ms
8,920 KB
1_test2.txt AC 18 ms
6,876 KB
1_test3.txt AC 18 ms
6,876 KB
1_test4.txt AC 18 ms
6,876 KB
1_test5.txt AC 17 ms
6,876 KB
1_test6.txt AC 233 ms
9,784 KB
1_test7.txt AC 244 ms
9,784 KB
1_test8.txt AC 417 ms
9,476 KB
1_test9.txt AC 424 ms
9,480 KB
1_test10.txt AC 431 ms
9,488 KB
1_test11.txt AC 399 ms
9,480 KB
1_test12.txt AC 582 ms
9,792 KB
1_test13.txt AC 527 ms
9,780 KB
1_test14.txt AC 603 ms
9,780 KB
1_test15.txt AC 566 ms
9,784 KB
1_test16.txt AC 19 ms
6,876 KB
1_test17.txt AC 19 ms
6,872 KB
1_test18.txt AC 19 ms
6,872 KB
1_test19.txt AC 19 ms
6,876 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