結果

問題 No.3046 White Tiger vs Monster
ユーザー gew1fw
提出日時 2025-06-12 16:20:34
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 884 bytes
コンパイル時間 387 ms
コンパイル使用メモリ 82,880 KB
実行使用メモリ 131,696 KB
最終ジャッジ日時 2025-06-12 16:20:55
合計ジャッジ時間 11,573 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 1 WA * 79
権限があれば一括ダウンロードができます

ソースコード

diff # 

import bisect

MOD = 10**9 + 7

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    K = int(input[ptr])
    ptr += 1
    N = int(input[ptr])
    ptr += 1
    x = list(map(int, input[ptr:ptr+N]))
    ptr += N
    x.sort()  # Ensure x is sorted, though input says it is
    
    dp = [0] * (K + 1)
    dp[0] = 1
    pre = [0] * (K + 1)
    pre[0] = dp[0]
    
    for i in range(1, K + 1):
        # Find the largest x_j <= i
        m = bisect.bisect_right(x, i) - 1
        if m < 0:
            s = 0
        else:
            s = 0
            for j in range(m + 1):
                k = i - x[j]
                if k >= 0:
                    s += dp[k]
                    if s >= MOD:
                        s -= MOD
        dp[i] = s % MOD
        pre[i] = (pre[i - 1] + dp[i]) % MOD
    
    print(dp[K] % MOD)

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