結果

問題 No.827 総神童数
ユーザー ks2m
提出日時 2019-05-04 00:50:35
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,081 ms / 2,000 ms
コード長 1,861 bytes
コンパイル時間 2,520 ms
コンパイル使用メモリ 81,260 KB
実行使用メモリ 94,216 KB
最終ジャッジ日時 2025-01-01 17:28:40
合計ジャッジ時間 26,661 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Queue;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		Map<Integer, List<Integer>> map = new HashMap<Integer, List<Integer>>();
		for (int i = 0; i < n - 1; i++) {
			String[] uv = br.readLine().split(" ");
			int u = Integer.parseInt(uv[0]);
			int v = Integer.parseInt(uv[1]);

			List<Integer> list1 = map.get(u);
			if (list1 == null) {
				list1 = new ArrayList<Integer>();
				map.put(u, list1);
			}
			list1.add(v);

			List<Integer> list2 = map.get(v);
			if (list2 == null) {
				list2 = new ArrayList<Integer>();
				map.put(v, list2);
			}
			list2.add(u);
		}
		br.close();

		if (n == 1) {
			System.out.println(1);
			return;
		}

		int[] dep = new int[n + 1];
		Queue<Integer> que = new ArrayDeque<Integer>();
		que.add(1);
		dep[1] = 1;
		while (!que.isEmpty()) {
			Integer cur = que.poll();
			List<Integer> list = map.get(cur);
			for (Integer next : list) {
				if (dep[next] == 0) {
					que.add(next);
					dep[next] = dep[cur] + 1;
				}
			}
		}

		int mod = 1000000007;
		long p = 1;
		for (int i = 1; i <= n; i++) {
			p *= i;
			p %= mod;
		}

		long ans = 0;
		for (int i = 1; i <= n; i++) {
			ans += p * modinv(dep[i], mod) % mod;
			ans %= mod;
		}
		System.out.println(ans);
	}

	static long modinv(long a, long m) {
		long b = m;
		long u = 1;
		long v = 0;
		long tmp = 0;

		while (b > 0) {
			long t = a / b;
			a -= t * b;
			tmp = a;
			a = b;
			b = tmp;

			u -= t * v;
			tmp = u;
			u = v;
			v = tmp;
		}

		u %= m;
		if (u < 0) u += m;
		return u;
	}
}
0