[BOJ] 백준 1967 트리의 지름 (Swift)

문제

https://www.acmicpc.net/problem/1967

1967번: 트리의 지름

풀이

https://dev-mandos.tistory.com/291

[BOJ] 백준 1167 트리의 지름 (Swift)

위 문제와 입력만 다르고 완전 똑같은 문제였습니다.

소스코드

	let n = Int(readLine()!)!
	var graph = [[(Int, Int)]](repeating: [], count: n + 1)
	var visited = [Bool](repeating: false, count: n + 1)
	for _ in 0..<n - 1{
	let input = readLine()!.split(separator: " ").map { Int($0)! }
	let a = input[0], b = input[1], cost = input[2]
	graph[a].append((b, cost))
	graph[b].append((a, cost))
	}
	var endNode: (node: Int, cost: Int) = (node: 0, cost: 0)
	func dfs(node: Int, depth: Int) {
	visited[node] = true
	if depth > endNode.cost {
	endNode = (node, depth)
	}
	for (nextNode, distance) in graph[node] {
	if !visited[nextNode] {
	visited[nextNode] = true
	dfs(node: nextNode, depth: depth + distance)
	visited[nextNode] = false
	}
	}
	}
	dfs(node: 1, depth: 0)
	visited = [Bool](repeating: false, count: n + 1)
	dfs(node: endNode.node, depth: 0)
	print(endNode.cost)

view raw 트리의 지름.swift hosted with ❤ by GitHub

후기

이전 문제를 풀었기에 쉽게 풀 수 있었던 문제였습니다.