結果

問題 No.3119 A Little Cheat
ユーザー titia
提出日時 2025-04-30 04:35:35
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,804 bytes
コンパイル時間 787 ms
コンパイル使用メモリ 82,888 KB
実行使用メモリ 98,688 KB
最終ジャッジ日時 2025-04-30 04:35:48
合計ジャッジ時間 12,016 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other TLE * 1 -- * 48
権限があれば一括ダウンロードができます

ソースコード

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

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

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


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


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

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

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

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

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


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

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

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

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


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


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

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

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


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

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

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

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


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

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

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

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


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

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

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


    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