結果

問題 No.3119 A Little Cheat
ユーザー titia
提出日時 2025-04-30 04:38:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 165 ms / 2,000 ms
コード長 3,032 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 102,272 KB
最終ジャッジ日時 2025-04-30 04:38:27
合計ジャッジ時間 9,749 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N,M=map(int,input().split())
A=list(map(int,input().split()))

mod=998244353

DP=[0,0,0,0]
DP[1]=min(A[0],A[1])
DP[2]=max(A[0],A[1])-min(A[0],A[1])
DP[3]=M-DP[1]-DP[2]

A.append(1)

for i in range(N-1):
    NDP=[0,0,0,0]
    NDP[0]+=DP[0]*M%mod

    if A[i]<A[i+1]:
        if A[i+1]<A[i+2]:
            NDP[0]+=DP[1]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[1]*A[i]%mod
            NDP[2]+=DP[1]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[1]*(M-A[i+2])%mod


            NDP[1]+=DP[2]*A[i+1]%mod
            NDP[2]+=DP[2]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[2]*(M-A[i+2])%mod


            NDP[0]+=DP[3]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[3]*A[i]%mod
            NDP[2]+=DP[3]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[3]*(M-A[i+2])%mod

        elif A[i]<A[i+2]<=A[i+1]:
            NDP[0]+=DP[1]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[1]*A[i]%mod
            NDP[3]+=DP[1]*(M-A[i+1])%mod

            NDP[1]+=DP[2]*A[i+2]%mod
            NDP[2]+=DP[2]*(A[i+1]-A[i+2])%mod
            NDP[3]+=DP[2]*(M-A[i+1])%mod


            NDP[0]+=DP[3]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[3]*A[i]%mod
            NDP[3]+=DP[3]*(M-A[i+1])%mod

        elif A[i+2]<=A[i]:
            NDP[0]+=DP[1]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[1]*A[i+2]%mod
            NDP[2]+=DP[1]*(A[i]-A[i+2])%mod
            NDP[3]+=DP[1]*(M-A[i+1])%mod


            NDP[1]+=DP[2]*A[i+2]%mod
            NDP[2]+=DP[2]*(A[i+1]-A[i+2])%mod
            NDP[3]+=DP[2]*(M-A[i+1])%mod


            NDP[0]+=DP[3]*(A[i+1]-A[i])%mod

            NDP[1]+=DP[3]*A[i+2]%mod
            NDP[2]+=DP[3]*(A[i]-A[i+2])%mod
            NDP[3]+=DP[3]*(M-A[i+1])%mod

    else:
        if A[i]<A[i+2]:
            NDP[1]+=DP[1]*A[i+1]%mod
            NDP[2]+=DP[1]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[1]*(M-A[i+2])%mod


            NDP[0]+=DP[2]*(A[i+1]+(M-A[i]))%mod

            NDP[2]+=DP[2]*(A[i]-A[i+1])%mod

            NDP[1]+=DP[3]*A[i+1]%mod
            NDP[2]+=DP[3]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[3]*(M-A[i+2])%mod

        elif A[i+1]<A[i+2]<=A[i]:
            NDP[1]+=DP[1]*A[i+1]%mod
            NDP[2]+=DP[1]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[1]*(M-A[i+2])%mod


            NDP[0]+=DP[2]*(A[i+1]+(M-A[i]))%mod

            NDP[2]+=DP[2]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[2]*(A[i]-A[i+2])%mod

            NDP[1]+=DP[3]*A[i+1]%mod
            NDP[2]+=DP[3]*(A[i+2]-A[i+1])%mod
            NDP[3]+=DP[3]*(M-A[i+2])%mod

        elif A[i+2]<=A[i+1]:
            NDP[1]+=DP[1]*A[i+2]%mod
            NDP[2]+=DP[1]*(A[i+1]-A[i+2])%mod
            NDP[3]+=DP[1]*(M-A[i+1])%mod


            NDP[0]+=DP[2]*(A[i+1]+(M-A[i]))%mod

            NDP[3]+=DP[2]*(A[i]-A[i+1])%mod

            NDP[1]+=DP[3]*A[i+2]%mod
            NDP[2]+=DP[3]*(A[i+1]-A[i+2])%mod
            NDP[3]+=DP[3]*(M-A[i+1])%mod


    DP=NDP
    DP[0]%=mod
    DP[1]%=mod
    DP[2]%=mod

    #print(DP)


ANS=0
k=pow(M,N-1,mod)

for i in range(N):
    ANS+=k*(M-A[i])%mod

print((ANS+DP[0])%mod)
0