結果

問題 No.412 花火大会
ユーザー tktk_snsn
提出日時 2020-12-08 20:29:06
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 727 bytes
コンパイル時間 130 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-09-18 18:08:37
合計ジャッジ時間 1,832 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

import bisect
bcd = sorted(map(int, input().split()))
n = int(input())
E = sorted(map(int, input().split()))

dp = [0] * 8
dp[0] = 1
for e in E:
    newDP = dp[:]
    b = bisect.bisect_right(bcd, e)
    if b == 0:
        newDP = [x * 2 for x in dp]
    elif b == 1:
        for bit in range(8):
            if bit & 1:
                newDP[bit] += dp[bit]
                newDP[bit] += dp[bit ^ 1]
    elif b == 2:
        newDP[2] += dp[0]
        newDP[3] += dp[1] + dp[2] + dp[3]
        newDP[6] += dp[4]
        newDP[7] += dp[5] + dp[6] + dp[7]
    else:
        newDP[4] += dp[0]
        newDP[5] += dp[1]
        newDP[6] += dp[2] + dp[4]
        newDP[7] += dp[3] + dp[5] + dp[6] + dp[7]
    dp = newDP

print(dp[7])
0