結果

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

ソースコード

diff #
raw source code 

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]].add(u[1])
 g[u[1]].add(u[0])
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