結果
問題 |
No.1294 マウンテン数列
|
ユーザー |
|
提出日時 | 2021-02-16 16:59:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 352 ms / 2,000 ms |
コード長 | 1,659 bytes |
コンパイル時間 | 190 ms |
コンパイル使用メモリ | 82,240 KB |
実行使用メモリ | 77,524 KB |
最終ジャッジ日時 | 2024-09-13 04:26:55 |
合計ジャッジ時間 | 3,804 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 17 |
ソースコード
def main3(n,a): mod=998244353 a.reverse() # a[0]>a[1]>..>a[n-2]>a[n-1] # a[0]が必ずマウンテンの頂点になる。 # a[1],a[2],a[3]と順番に数列に追加していく。追加する要素は数列の右端または左端に追加する。 # a[i]を追加するとき、両端のうち一つはa[i-1]になっている。もう片方の端の値で状態をまとめる。 # dp[j]:a[i]での遷移を考えているとき、片方の端の値がa[i-1]でもう片方がa[j]である場合数。 # 差がd以下で遷移すれば、危険度d以下のマウンテンの個数がわかる。 # 後は包除原理で総和を求める。 mxd=a[0]-a[-1] # 最大の危険度 mnd=mxd # 最小の危険度 for x,y in zip(a,a[1:]):mnd=min(mnd,x-y) ary=[0]*(mxd+1) # ary[d]:危険度dのマウンテン数列の個数 for d in range(mnd,mxd+1): # dp0=[1] # dp0[j]:a[i]の遷移を考えているとき、両端のa[i-1]じゃない方がa[j]である場合数 if a[0]-a[1]>d:continue dp1=[0,1] # dp1:dp0の累積和。dp0は使わない。 idx=0 flg=True for i in range(2,n): # a[i]での遷移を考える。 while idx<n and a[idx]-a[i]>d:idx+=1 if idx==i: flg=False break # a[i]はa[k](idx<=k<=i-1)の隣における。 t=dp1[-1]-dp1[idx] t%=mod dp1.append((dp1[-1]+t)%mod) if not flg:continue ary[d]=dp1[-1] ary[d]%=mod ans=0 for d in range(mnd,mxd+1): ans+=d*(ary[d]-ary[d-1]) ans%=mod return ans*2%mod if __name__=='__main__': n=int(input()) a=list(map(int,input().split())) ret3=main3(n,a) print(ret3)