結果

問題 No.2598 Kadomatsu on Tree
ユーザー 👑 seekworser
提出日時 2024-01-03 04:48:34
言語 Nim
(2.2.0)
結果
RE  
実行時間 -
コード長 630 bytes
コンパイル時間 5,053 ms
コンパイル使用メモリ 65,888 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-09-27 18:23:19
合計ジャッジ時間 9,884 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 60
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(24, 8) Warning: Number of spaces around '!=' is not consistent [Spacing]

ソースコード

diff # 

import strutils,sequtils
var m=998244353
var n=stdin.readLine.parseint
var g=newSeqWith(n,newSeq[int](0))
for i in 0..<n-1:
 var u=stdin.readline.split.map(parseint).mapit(it-1)
 g[u[0]-1].add(u[1]-1)
 g[u[1]-1].add(u[0]-1)
var a=stdin.readLine.split.map(parseInt)
var ans=0
proc dfs(x,par:int):int=
 result=1
 var tl,tu=0
 for y in g[x]:
  if y==par: continue
  var sz=dfs(y,x)
  result+=sz
  if a[y]<a[x]:
   ans+=(tl*sz)mod m
   tl+=sz
  if a[x]<a[y]:
   ans+=(tu*sz)mod m
   tu+=sz
 if par!= -1:
  if a[par]<a[x]: ans+=(tl*(n-result))mod m
  if a[x]<a[par]: ans+=(tu*(n-result))mod m
  ans=ans mod m
discard dfs(0,-1)
echo ans
0