結果

問題 No.2653 [Cherry 6th Tune] Re: start! (Make it Zero!)
ユーザー 👑 KazunKazun
提出日時 2023-12-08 01:53:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 283 ms / 2,000 ms
コード長 1,300 bytes
コンパイル時間 364 ms
コンパイル使用メモリ 81,884 KB
実行使用メモリ 105,820 KB
最終ジャッジ日時 2024-09-29 04:32:39
合計ジャッジ時間 14,848 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    N, M = map(int, input().split())
    X = list(map(int, input().split()))
    Mod = 998244353

    Q = X.count(-1)
    if Q == 1:
        i = X.index(-1)
        X[i] = (- sum(X[j] for j in range(N) if j != i)) % M

    if -1 not in X:
        if sum(X) % M != 0:
            return 0

        cum = 0
        k = 0
        for x in X:
            cum += x; cum %= M
            if cum == 0:
                k += 1
        return N - k

    left = min(i for i in range(N) if X[i] == -1)
    right = max(i for i in range(N) if X[i] == -1)

    cum = 0
    S = 0
    M_pow_Q1 = pow(M, Q - 1, Mod); M_pow_Q2 = pow(M, Q - 2, Mod)

    for i in range(N):
        if i < left:
            cum += X[i]; cum %= M
            if cum == 0:
                S += M_pow_Q1
        elif left <= i < right:
            S += M_pow_Q2
        else:
            if i == right:
                cum = (M - sum(X[i] for i in range(right + 1, N))) % M
            else:
                cum += X[i]
                cum %= M

            if cum == 0:
                S += M_pow_Q1

    return (N * M_pow_Q1 - S) % Mod

#==================================================
import sys
input = sys.stdin.readline
write = sys.stdout.write

T = int(input())
write("\n".join(map(str, [solve() for _ in range(T)])))
0