結果

問題 No.277 根掘り葉掘り
ユーザー hama_duhama_du
提出日時 2015-11-20 19:13:20
言語 Go
(1.22.1)
結果
AC  
実行時間 242 ms / 3,000 ms
コード長 1,998 bytes
コンパイル時間 11,073 ms
コンパイル使用メモリ 237,784 KB
実行使用メモリ 27,644 KB
最終ジャッジ日時 2024-10-10 20:12:03
合計ジャッジ時間 14,543 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 AC 1 ms
5,248 KB
testcase_04 AC 1 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 AC 1 ms
5,248 KB
testcase_08 AC 1 ms
5,248 KB
testcase_09 AC 242 ms
27,644 KB
testcase_10 AC 198 ms
16,640 KB
testcase_11 AC 206 ms
15,872 KB
testcase_12 AC 213 ms
18,304 KB
testcase_13 AC 206 ms
15,872 KB
testcase_14 AC 215 ms
16,252 KB
testcase_15 AC 211 ms
17,024 KB
testcase_16 AC 208 ms
14,152 KB
testcase_17 AC 210 ms
14,592 KB
testcase_18 AC 209 ms
13,824 KB
testcase_19 AC 206 ms
16,128 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import (
	"fmt"
	"bufio"
	"os"
	"strconv"
	"io"
)

var depth []int
var rdepth []int
var graph [][]int
const INF = 114514

func dfs(now, par, dep int) int {
	depth[now] = dep
	min := INF
	for i := 0 ; i < len(graph[now]) ; i++ {
		to := graph[now][i]
		if to == par {
			continue
		}
		cn := dfs(to, now, dep+1) + 1
		if min > cn {
			min = cn
		}
	}
	if min == INF {
		min = 0
	}
	rdepth[now] = min
	return min
}

func dfs2(now, par int) {
	if par >= 0 && rdepth[now] > rdepth[par] + 1 {
		rdepth[now] = rdepth[par] + 1
	}
	for i := 0 ; i < len(graph[now]) ; i++ {
		to := graph[now][i]
		if to == par {
			continue
		}
		dfs2(to, now)
	}
}

func main() {
	n := nextInt()
	edges := make([][]int, n-1)
	deg := make([]int, n)
	for i := 0 ; i < n-1 ; i++ {
		edges[i] = make([]int, 2)
		a := nextInt()-1
		b := nextInt()-1
		deg[a]++
		deg[b]++
		edges[i][0] = a
		edges[i][1] = b
	}
	graph = make([][]int, n)
	for i := 0 ; i < n ; i++ {
		graph[i] = make([]int, deg[i])
	}
	for i := 0 ; i < n-1 ; i++ {
		a := edges[i][0]
		b := edges[i][1]
		deg[a]--
		deg[b]--
		graph[a][deg[a]] = b
		graph[b][deg[b]] = a
	}

	depth = make([]int, n)
	rdepth = make([]int, n)
	dfs(0, -1, 0)
	dfs2(0, -1)

	for i := 0 ; i < n ; i++ {
		if depth[i] < rdepth[i] {
			fmt.Println(depth[i])
		} else {
			fmt.Println(rdepth[i])
		}
	}
}

// ====

var rdr = bufio.NewReader(os.Stdin)

func nextInt() int {
	i, e := strconv.Atoi(readToken(20))
	if e != nil {
		panic(e)
	}
	return i
}

func nextString(limit int) string {
	return readToken(limit)
}

func readToken(limit int) string {
	buf := make([]byte, 0, limit)

	for {
		byte, err := rdr.ReadByte()
		if err != nil {
			if err == io.EOF {
				break
			}
		}
		if byte != 10 && byte != 13 && byte != 32 {
			buf = append(buf, byte)
			break
		}
	}
	for {
		byte, err := rdr.ReadByte()
		if err != nil {
			if err == io.EOF {
				break
			}
		}
		if byte == 10 || byte == 13 || byte == 32 {
			break
		}
		buf = append(buf, byte)
	}
	return string(buf)
}
0