結果

問題 No.1736 Princess vs. Dragoness
ユーザー Theta
提出日時 2022-10-26 19:09:12
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,109 bytes
コンパイル時間 122 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,392 KB
最終ジャッジ日時 2024-07-04 07:52:42
合計ジャッジ時間 2,451 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_right
from itertools import accumulate
from math import ceil


def main():
    N, A, B, X, Y = map(int, input().split())
    H = list(map(int, input().split()))
    H_partial_sum = list(accumulate(H))
    used_num_of_magic_A = list(map(lambda num: ceil(num // X), H))
    sum_used_num_of_magic_A_reversed = list(accumulate(
        reversed(used_num_of_magic_A)))

    magic_B_kill_idx = bisect_right(H_partial_sum, Y*B)
    magic_A_kill_idx = N - bisect_right(sum_used_num_of_magic_A_reversed, A)

    if magic_B_kill_idx >= magic_A_kill_idx:
        print("Yes")
        return

    if magic_B_kill_idx + 1 < magic_A_kill_idx:
        print("No")
        return
    try:
        magic_B_rest = Y*B - H_partial_sum[magic_B_kill_idx-1]
    except IndexError:
        magic_B_rest = Y*B
    try:
        magic_A_rest_num = A - used_num_of_magic_A[N - magic_A_kill_idx - 1]
    except IndexError:
        magic_A_rest_num = A

    if magic_B_rest + magic_A_rest_num * X >= H[magic_B_kill_idx]:
        print("Yes")
    else:
        print("No")


if __name__ == "__main__":
    main()
0