結果

問題 No.1049 Zero (Exhaust)
ユーザー dot_haraai
提出日時 2021-05-27 22:30:19
言語 Nim
(2.2.0)
結果
AC  
実行時間 100 ms / 2,000 ms
コード長 1,193 bytes
コンパイル時間 4,849 ms
コンパイル使用メモリ 86,400 KB
実行使用メモリ 50,688 KB
最終ジャッジ日時 2024-11-06 16:08:03
合計ジャッジ時間 6,831 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(2, 18) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(1, 8) Warning: imported and not used: 'times' [UnusedImport]
/home/judge/data/code/Main.nim(2, 26) Warning: imported and not used: 'strformat' [UnusedImport]
/home/judge/data/code/Main.nim(2, 18) Warning: imported and not used: 'future' [UnusedImport]
/home/judge/data/code/Main.nim(2, 37) Warning: imported and not used: 'deques' [UnusedImport]
/home/judge/data/code/Main.nim(1, 41) Warning: imported and not used: 'algorithm' [UnusedImport]
/home/judge/data/code/Main.nim(1, 52) Warning: imported and not used: 'tables' [UnusedImport]
/home/judge/data/code/Main.nim(1, 60) Warning: imported and not used: 'sets' [UnusedImport]
/home/judge/data/code/Main.nim(1, 66) Warning: imported and not used: 'lists' [UnusedImport]
/home/judge/data/code/Main.nim(1, 73) Warning: imported and not used: 'intsets' [UnusedImport]
/home/judge/data/code/Main.nim(2, 8) Warning: imported and not used: 'critbits' [UnusedImport]

ソースコード

diff # 

import times, strutils, sequtils, math, algorithm, tables, sets, lists, intsets
import critbits, future, strformat, deques
template `max=`(x,y) = x = max(x,y)
template `min=`(x,y) = x = min(x,y)
template `mod=`(x,y) = x = x mod y
template scan2 = (scan(), scan())
template scan3 = (scan(), scan())
let read* = iterator: string {.closure.} =
    while true: (for s in stdin.readLine.split: yield s)
proc scan(): int = read().parseInt
proc scanf(): float = read().parseFloat
proc toInt(c:char): int =
    return int(c) - int('0')




proc solve():int=
  var
    p = scan()
    k = scan()
    dp = newseqwith(k+1,newseqwith(2,0))
    md = 10^9+7

  # N = 0のとき、次が0になる
  # 掛け算：0~(p-1)のp通り
  # 足し算：0の1通り
  # 次が0以外になる
  # 掛け算：0通り
  # 足し算：p-1通り
  # N != 0のとき、次が0になる
  # 掛け算：0の1通り
  # 足し算：1通り
  # 次が0以外になる
  # 掛け算：p-1通り
  # 足し算：p-1通り
  dp[0][0]=1
  for v in 1..k:
    dp[v][0] = (dp[v-1][0]*(p+1) + dp[v-1][1]*2).mod(md)
    dp[v][1] = (dp[v-1][0]*(p-1)+dp[v-1][1]*(p-1)*2).mod(md)
  #echo dp.join("\n")
  return dp[k][0]


echo solve()
0