結果

問題 No.1207 グラフX
ユーザー dot_haraaidot_haraai
提出日時 2021-05-31 20:33:59
言語 Nim
(2.2.0)
結果
AC  
実行時間 506 ms / 2,000 ms
コード長 1,798 bytes
コンパイル時間 4,075 ms
コンパイル使用メモリ 88,192 KB
実行使用メモリ 72,448 KB
最終ジャッジ日時 2024-11-08 21:37:24
合計ジャッジ時間 23,672 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます
コンパイルメッセージ
/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, 37) Warning: imported and not used: 'deques' [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(2, 18) Warning: imported and not used: 'future' [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 toInt(c:char): int =
    return int(c) - int('0')




proc modPow(x,n,md:int):int=
  var
    x = x
    n = n
  result = 1
  while n > 0:
    if (n and 1) > 0:
      result*=x
      result.mod=md
    x = (x^2).mod(md)
    n = n shr 1




proc solve()=
  var
    n = scan()
    m = scan()
    X = scan()
    #
    xyz = newseqwith(m,(scan()-1,scan()-1,scan()))
    md = 10^9+7
    roots = newseqwith(n,0)
  xyz = xyz.sortedByIt(it[2])
  
  for i in 0..<n:
    roots[i] = i
  
  proc find(x:int):int=
    if roots[x] == x:
      return x
    else:
      result = find(roots[x])
      roots[x] = result

  proc unite(x,y:int)=
    var
      rx = find(x)
      ry = find(y)
    if rx!=ry:
      roots[ry] = rx
    return

  var
    es = newseqwith(n,newseq[int]())
    cs = newseqwith(n,newseq[int]())
    res = 0
  
  for (x,y,z) in xyz:
    if find(x)!=find(y):
      es[x].add(y)
      es[y].add(x)
      cs[x].add(z)
      cs[y].add(z)
      unite(x,y)
  #echo es.join("\n")
  #echo ""
  #echo cs.join("\n")
  #echo  roots
  
  proc dfs(par,p:int):int=
    result = 1
    for (nxt,cost) in zip(es[p],cs[p]):
      if nxt == par:continue
      var t = dfs(p,nxt)
      res+=(((n-t)*t).mod(md)*modPow(X,cost,md)).mod(md)
      res.mod=md
      result+=t
  discard dfs(-1,0)
  echo res
    

  

  
  
    
solve()
0