ABC307 E問題解説(Python)

E問題 Distinct Adjacent提出コード mod=998244353n,m=map(int,input().split())dp=[[0,0] for i in range(n)]dp[0][1]=1 for i in range(1,n): dp[i][0]+=dp[i-1][0]*(m-2)%mod dp[i][0]+=dp[i-1][1]*(m-1)%mod dp[i][1]+=dp[i-1][0]%mod print(dp[n-1][0