結果

問題 No.827 総神童数
ユーザー dot_haraai
提出日時 2021-06-13 19:21:26
言語 Nim
(2.2.0)
結果
AC  
実行時間 221 ms / 2,000 ms
コード長 1,700 bytes
コンパイル時間 3,636 ms
コンパイル使用メモリ 88,680 KB
実行使用メモリ 16,896 KB
最終ジャッジ日時 2024-12-23 08:49:39
合計ジャッジ時間 9,807 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
/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, 18) Warning: imported and not used: 'future' [UnusedImport]
/home/judge/data/code/Main.nim(2, 26) Warning: imported and not used: 'strformat' [UnusedImport]
/home/judge/data/code/Main.nim(2, 44) Warning: imported and not used: 'heapqueue' [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,heapqueue
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 scanbf(): BiggestFloat = read().parseFloat
proc toInt(c:char): int =
    return int(c) - int('0')

proc xgcd(a,b:int):(int,int,int)=
  var
    b = b
    x0=1
    y0=0
    x1=0
    y1=1
    q:int
    a = a
  while b!=0:
    (q,a,b) = (a.div(b),b,a.mod(b))
    (x0,x1)=(x1,x0-q*x1)
    (y0,y1)=(y1,y0-q*y1)
  return (a,x0,y0)

# modの逆元
proc modinv(a,m:int):int=
  var (g,x,y)=xgcd(a,m)
  if g!=1:
    return -1
  else:
    return ((x mod m)+m).mod(m)

    
proc solve():int=
  var
    n = scan()
    es = newseqwith(n,newseq[int]())
    fact = newseqwith(n+1,1)
    depth = newseqwith(n,int.high.div(4))
    minv = newseqwith(n+1,0)
    md = 10^9+7
  
  for i in 0..<n-1:
    var
      (u,v) = (scan()-1,scan()-1)
    es[u].add(v)
    es[v].add(u)

  fact[1]=1
  for i in 2..n:
    fact[i]=(fact[i-1]*i).mod(md)
  for i in 1..n:
    minv[i] = modinv(i,md)
  
  var
    q = initDeque[int]()
  q.addLast(0)
  depth[0]=1
  while q.len>0:
    var p = q.popFirst
    for nxt in es[p]:
      if depth[nxt]>depth[p]+1:
        depth[nxt]=depth[p]+1
        q.addLast(nxt)
  #echo depth
  for i in 0..<n:
    result+=fact[n]*minv[depth[i]]
    result.mod=md
  result = (result.mod(md)+md).mod(md)

  
  
echo solve()
0