提出 #46633862

ソースコード 拡げる

def minimum_unbalancedness(N, M, toasts):
    # Sort the toasts' deliciousness in decreasing order
    toasts.sort(reverse=True)

    # The first 2M-N toasts should be on their own plates
    unbalancedness = sum(x * x for x in toasts[:2*M-N])
    
    # For the remaining toasts, pair them by pairing the largest remaining with the smallest remaining and so on
    for i in range(2*M-N, 2*M-N + (N - M)):
        paired_value = toasts[i] + toasts[N - (i - (2*M - N)) - 1]
        unbalancedness += paired_value**2

    return unbalancedness

# Read input
N, M = map(int, input().split())
toasts = list(map(int, input().split()))

# Print answer
print(minimum_unbalancedness(N, M, toasts))

提出情報

提出日時
問題 A - Toasts for Breakfast Party
ユーザ scott_wu
言語 Python (PyPy 3.10-v7.3.12)
得点 300
コード長 708 Byte
結果 AC
実行時間 106 ms
メモリ 115408 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 300 / 300
結果
AC × 3
AC × 25
セット名 テストケース
Sample 01-sample-01.txt, 01-sample-02.txt, 01-sample-03.txt
All 01-sample-01.txt, 01-sample-02.txt, 01-sample-03.txt, 02-min-01.txt, 03-max-01.txt, 04-random-01.txt, 04-random-02.txt, 04-random-03.txt, 04-random-04.txt, 04-random-05.txt, 05-large-rand-01.txt, 05-large-rand-02.txt, 05-large-rand-03.txt, 05-large-rand-04.txt, 05-large-rand-05.txt, 05-large-rand-06.txt, 05-large-rand-07.txt, 05-large-rand-08.txt, 05-large-rand-09.txt, 05-large-rand-10.txt, 06-sukunai-01.txt, 06-sukunai-02.txt, 06-sukunai-03.txt, 06-sukunai-04.txt, 06-sukunai-05.txt
ケース名 結果 実行時間 メモリ
01-sample-01.txt AC 54 ms 76628 KiB
01-sample-02.txt AC 54 ms 76632 KiB
01-sample-03.txt AC 54 ms 76668 KiB
02-min-01.txt AC 54 ms 76748 KiB
03-max-01.txt AC 79 ms 114304 KiB
04-random-01.txt AC 72 ms 89828 KiB
04-random-02.txt AC 83 ms 98496 KiB
04-random-03.txt AC 61 ms 81728 KiB
04-random-04.txt AC 98 ms 109500 KiB
04-random-05.txt AC 67 ms 86596 KiB
05-large-rand-01.txt AC 104 ms 114700 KiB
05-large-rand-02.txt AC 105 ms 115128 KiB
05-large-rand-03.txt AC 104 ms 114216 KiB
05-large-rand-04.txt AC 104 ms 114508 KiB
05-large-rand-05.txt AC 104 ms 114832 KiB
05-large-rand-06.txt AC 106 ms 115288 KiB
05-large-rand-07.txt AC 105 ms 114824 KiB
05-large-rand-08.txt AC 106 ms 115408 KiB
05-large-rand-09.txt AC 104 ms 114544 KiB
05-large-rand-10.txt AC 105 ms 114192 KiB
06-sukunai-01.txt AC 72 ms 97512 KiB
06-sukunai-02.txt AC 77 ms 100476 KiB
06-sukunai-03.txt AC 68 ms 89660 KiB
06-sukunai-04.txt AC 60 ms 81932 KiB
06-sukunai-05.txt AC 72 ms 94248 KiB