結果

問題 No.2638 Initial fare
ユーザー ID 21712ID 21712
提出日時 2024-11-11 15:52:18
言語 Go
(1.23.4)
結果
WA  
実行時間 -
コード長 578 bytes
コンパイル時間 13,297 ms
コンパイル使用メモリ 222,848 KB
実行使用メモリ 64,640 KB
最終ジャッジ日時 2024-11-11 15:52:39
合計ジャッジ時間 20,075 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample WA * 3
other WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import . "fmt"
import . "os"
import bf "bufio"

func main() {
	rd:=bf.NewReader(Stdin)
	var n int
	Fscan(rd,&n)
	tree:=make([][]int,n+1)
	for e:=0;e<n-1;e++ {
		var u,v int
		Fscan(rd,&u,&v)
		tree[u]=append(tree[u],v)
		tree[v]=append(tree[v],u)
	}
	visited:=make([]bool,n+1)
	var ans int
	var dfs func(t int) (a,b,c int)
	dfs = func(t int) (a,b,c int){
		visited[t]=true
		a=1
		for _,k:=range tree[t] {
			if visited[k] {
				continue
			}
			x,y,z:=dfs(k)
			ans+=(a+b+c)*x+(a+b)*y+a*z
			b+=x
			c+=y
		}
		return
	}
	x,y,z:=dfs(1)
	ans+=x+y+z
	Println(ans)
}
0