Submission #17182584

---

Source Code Expand

import Foundation
// 第一引数には隣接リストで表したグラフを渡す (参照渡し)
// 第二引数には探索をスタートする頂点を渡す
// 第三引数には x の情報をまとめたリストを渡す (参照渡し)
// 第四引数には答えを保存するリストを渡す (参照渡し)
// 第五引数にはすでに訪れた頂点を保存しておくリストを渡す (参照渡し)
// 第六変数にはカウンターに記録する値となるものを渡す
func dfs(graph:inout [[Int]], start:Int,plus:inout [Int],ans:inout [Int],visited:inout [Bool],num:Int){
    // 訪問済みの頂点を記録する
    visited[start] = true
    // カウンターに値を追加
    ans[start] = num+plus[start]
    // 隣接する頂点についてループを回す
    for i in graph[start]{
        // すでに訪れていたらcontinue
        if visited[i]{
            continue
        }
        // 再帰
        dfs(graph:&graph,start:i,plus:&plus,ans:&ans,visited:&visited,num:num+plus[start])
    }
    
}
// 標準入力の受け取り
let nq = readLine()!.split(separator:" ").map{Int($0)!}
let n = nq[0]
let q = nq[1]
var G:[[Int]] = []
for _ in 0..<n{
    G.append([])
}
// 木は隣接リストの形で受け取る
for _ in 0..<n-1{
    let ab = readLine()!.split(separator:" ").map{Int($0)!}
    let a = ab[0]
    let b = ab[1]
    G[a-1].append(b-1)
    G[b-1].append(a-1)
}
// 必要な変数の初期化
var plus:[Int] = []
for _ in 0..<n{
    plus.append(0)
}
// xiもリストの形でまとめて値を持っておく
for _ in 0..<q{
    let qx = readLine()!.split(separator:" ").map{Int($0)!}
    let q = qx[0]
    let x = qx[1]
    plus[q-1] += x
}
// 必要な変数の初期化
var ans:[Int] = []
var check:[Int] = []
var visited:[Bool] = []
for _ in 0..<n{
    ans.append(0)
    check.append(0)
    visited.append(false)
}
// dfs
dfs(graph:&G,start:0,plus:&plus,ans:&ans,visited:&visited,num:0)
// 文字列のリストに変換
var answer:[String] = ans.map{String($0)}
// リストを結合して一つの文字列にして出力
print(answer.joined(separator: " "))

Submission Info

Judge Result