結果

問題 No.827 総神童数
ユーザー matsukin1111
提出日時 2019-05-27 10:42:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 207 ms / 2,000 ms
コード長 1,614 bytes
コンパイル時間 862 ms
コンパイル使用メモリ 98,672 KB
実行使用メモリ 27,620 KB
最終ジャッジ日時 2024-09-17 15:28:30
合計ジャッジ時間 5,848 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<cstdio>
#include<cstring>
#include <cstdlib>  
#include <math.h>
#include <cmath>
#include<cctype>
#include<string>
#include<set>
#include<iomanip>
#include <map>
#include<algorithm>
#include <functional>
#include<vector>
#include<climits>
#include<stack>
#include<queue>
#include <deque>
#include <climits>
#include <typeinfo>
#include <utility> 
#define all(x) (x).begin(),(x).end()
#define rep(i,m,n) for(int i = m;i < n;++i)
#define pb push_back
#define fore(i,a) for(auto &i:a)
#define rrep(i,m,n) for(int i = m;i >= n;--i)
#define INF INT_MAX/2
using namespace std;
using ll = long long;
using R = double;
using Data = pair<ll, vector<int>>;
const ll MOD = 1e9 + 7;
const ll inf = 1LL << 50;
struct edge { ll from; ll to; ll cost; };
typedef tuple<ll, ll, ll>T;
typedef pair<ll,ll>pll;

int n;
vector<int>e[201010];
ll ans = 0;

const int MAX = 201010;
ll fac[MAX], finv[MAX], inv[MAX];
void combinit() {
	fac[0] = fac[1] = 1;
	finv[0] = finv[1] = 1;
	inv[1] = 1;
	for (int i = 2; i < MAX; i++) {
		fac[i] = fac[i - 1] * i % MOD;
		inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD;
		finv[i] = finv[i - 1] * inv[i] % MOD;
	}
}
ll comb(int n, int k) {
	if (n < k) return 0;
	if (n < 0 || k < 0) return 0;
	return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

void dfs(int cu,int pa,int dpt) {
	ans += (((comb(n,dpt)*fac[dpt-1])%MOD)*fac[n-dpt])%MOD;
	ans %= MOD;
	fore(to, e[cu])if (to != pa)dfs(to,cu,dpt+1);
}

int main(){
	cin >> n;
	combinit();
	rep(i, 0, n - 1) {
		int a, b;
		cin >> a >> b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(1,-1,1);
	cout << ans%MOD << endl;

	return 0;
}
0