結果

問題 No.2653 [Cherry 6th Tune] Re: start! (Make it Zero!)
コンテスト
ユーザー 👑 Kazun
提出日時 2023-06-25 23:31:59
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 272 ms / 2,000 ms
コード長 1,297 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 417 ms
コンパイル使用メモリ 85,316 KB
実行使用メモリ 111,172 KB
最終ジャッジ日時 2026-04-15 15:16:24
合計ジャッジ時間 11,929 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

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

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

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

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

    cum = 0
    H = (M - sum(A[i] for i in range(N) if A[i] != -1 and not(left < i < right))) % M
    S = 0

    for i in range(N):
        if i < left:
            cum += A[i]; cum %= M
            if cum == 0:
                S += pow(M, K-1, Mod)
        elif left <= i < right:
            S += pow(M, K-2, Mod)
        else:
            if i == right:
                cum += H
            else:
                cum += A[i]
            cum %= M
            if cum == 0:
                S += pow(M, K-1, Mod)

    return (N * pow(M, K-1, Mod) - 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