結果

問題 No.2807 Have Another Go (Easy)
ユーザー gew1fw
提出日時 2025-06-12 18:58:14
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,320 bytes
コンパイル時間 209 ms
コンパイル使用メモリ 82,676 KB
実行使用メモリ 271,924 KB
最終ジャッジ日時 2025-06-12 18:58:19
合計ジャッジ時間 4,699 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 TLE * 1 -- * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    N = int(input[ptr])
    ptr +=1
    M = int(input[ptr])
    ptr +=1
    k = int(input[ptr])
    ptr +=1
    Cs = list(map(int, input[ptr:ptr+k]))
    ptr +=k
    
    max_s = 2*N -1
    ways = [0]*(max_s +1)
    ways[0] =1
    for s in range(1, max_s+1):
        for d in range(1,7):
            if s -d >=0:
                ways[s] = (ways[s] + ways[s-d]) % MOD
    
    S =0
    for s in range(max_s+1):
        need = 2*N -s
        if need <=6:
            cnt = 7 - need
            S = (S + ways[s] * cnt) % MOD
    
    inv6 = pow(6, MOD-2, MOD)
    
    for c in Cs:
        r = c % N
        tdp = [0]*(max_s +1)
        tdp[0] =1 if 0 % N != r else 0
        for s in range(1, max_s +1):
            if s % N == r:
                tdp[s] =0
                continue
            total =0
            for d in range(1,7):
                if s -d >=0:
                    total = (total + tdp[s-d]) % MOD
            tdp[s] = total
        
        T =0
        for s in range(max_s +1):
            need = 2*N -s
            if need <=6:
                cnt =7 - need
                T = (T + tdp[s] * cnt) % MOD
        
        ans = (S - T) % MOD
        print(ans)
        
if __name__ == '__main__':
    main()
0