結果

問題 No.801 エレベーター
ユーザー persimmon-persimmon
提出日時 2021-06-29 15:11:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,131 ms / 2,000 ms
コード長 1,067 bytes
コンパイル時間 245 ms
コンパイル使用メモリ 81,752 KB
実行使用メモリ 82,304 KB
最終ジャッジ日時 2024-06-25 17:56:40
合計ジャッジ時間 19,097 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=sys.stdin.readline
def main2(n,m,k,lr):
  mod=10**9+7
  dp=[0]*(n+1)
  dp[1]=1
  now=0
  cnt_ary=[0]*(n+1)
  for l,r in lr:
    cnt_ary[l]+=1
    if r!=n:cnt_ary[r+1]-=1
  for i in range(n):cnt_ary[i+1]+=cnt_ary[i]
  for _ in range(k):
    ndp=[0]*(n+1)
    sdp=[0]
    for x in dp:sdp.append((sdp[-1]+x)%mod)
    mat0=[0]*(n+1)
    mat1=[0]*(n+1)
    for l,r in lr:
      mat0[r]+=sdp[r+1]-sdp[l]
      mat1[l]+=sdp[r+1]-sdp[l]

    # 下からの遷移。同じ階の移動も含む
    for j in range(1,n+1):
      # j階への遷移
      now+=cnt_ary[j]*dp[j]
      now%=mod
      ndp[j]+=now
      ndp[j]%=mod
      now-=mat0[j]
      now%=mod
    # 上からの遷移。同じ階の移動は含まない
    for j in reversed(range(1,n+1)):
      # j階への遷移
      ndp[j]+=now
      ndp[j]%=mod
      now+=cnt_ary[j]*dp[j]
      now-=mat1[j]
      now%=mod
    dp=ndp
  return dp[n]

if __name__=='__main__':
  n,m,k=map(int,input().split())
  lr=[list(map(int,input().split())) for _ in range(m)]
  ret2=main2(n,m,k,lr)
  print(ret2)
0