#434675 (Rust) No.806 木を道に

• 問題: No.806 木を道に
• ユーザー: yoma
• 提出日時: 2020-02-26 00:23:53
• 言語: Rust (1.77.0 + proconio)
• 結果: WA
• 実行時間: -
• コード長: 1,883 bytes
• コンパイル時間: 14,540 ms
• コンパイル使用メモリ: 377,844 KB
• 実行使用メモリ: 16,896 KB
• 最終ジャッジ日時: 2024-10-13 14:42:17
• 合計ジャッジ時間: 14,247 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	0 ms; 5,248 KB
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	AC	13 ms; 11,776 KB
testcase_25	AC	21 ms; 16,896 KB
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-

ソースコード

#[allow(unused_macros)]
macro_rules! input {
	(source = $s:expr, $($r:tt)*) => {
		let mut iter = $s.split_whitespace();
		input_inner!{iter, $($r)*}
	};
	($($r:tt)*) => {
		let s = {
			use std::io::Read;
			let mut s = String::new();
			std::io::stdin().read_to_string(&mut s).unwrap();
			s
		};
		let mut iter = s.split_whitespace();
		input_inner!{iter, $($r)*}
	};
}
#[allow(unused_macros)]
macro_rules! input_inner {
	($iter:expr) => {};
	($iter:expr, ) => {};

	($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
		let $var = read_value!($iter, $t);
		input_inner!{$iter $($r)*}
	};
}
#[allow(unused_macros)]
macro_rules! read_value {
	($iter:expr, ( $($t:tt),* )) => {
		( $(read_value!($iter, $t)),* )
	};

	($iter:expr, [ $t:tt ; $len:expr ]) => {
		(0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
	};

	($iter:expr, chars) => {
		read_value!($iter, String).chars().collect::<Vec<char>>()
	};

	($iter:expr, usize1) => {
		read_value!($iter, usize) - 1
	};

	($iter:expr, $t:ty) => {
		$iter.next().unwrap().parse::<$t>().expect("Parse error")
	};
}

fn dfs(s: usize, graph:&Vec<Vec<usize>>, used: &mut Vec<bool>, d: &mut Vec<i32>, cd: i32) {
	used[s] = true;
	d[s] = cd;
	for g in &graph[s] {
		if !used[*g] {
			dfs(*g, graph, used, d, cd+1);
		}
	}
}

fn main() {
	input! {
		n: usize,
		edges: [(usize1, usize1); (n-1)],
	}
	let mut graph: Vec<Vec<usize>> = vec![vec![]; n];
	let mut used: Vec<bool> = vec![false; n];
	let mut d: Vec<i32> = vec![0; n];
	let mut deg: Vec<usize> = vec![0; n];
	let mut dis = vec![];
	for (s, v) in &edges {
		deg[*s] += 1;
		deg[*v] += 1;
		graph[*s].push(*v);
		graph[*v].push(*s);
	}
	dfs(0, &graph, &mut used, &mut d, 0);
	for i in 0..n {
		if deg[i] == 1 {
			dis.push(d[i]);
		}
	}
	dis.sort_by(|a, b| b.cmp(a));
	let ans_ex = dis[0]+dis[1]+1;
	let ans = n as i32-ans_ex;
	println!("{}", ans);
	// println!("{:?}", d);
}