結果

問題 No.827 総神童数
ユーザー ok
提出日時 2019-05-03 23:52:58
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 1,909 bytes
コンパイル時間 958 ms
コンパイル使用メモリ 76,108 KB
実行使用メモリ 29,372 KB
最終ジャッジ日時 2024-12-31 20:26:31
合計ジャッジ時間 4,678 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<iomanip>
#include<cmath>
#include<vector>
#include<algorithm>

using namespace std;

#define int long long
#define rep(i,n) for(int i = 0; i < (n); i++)
#define INF ((long long)1e18)
#define MOD ((int)1e9+7)
#define endl "\n"

#define yn(f) ((f)?"Yes":"No")
#define YN(f) ((f)?"YES":"NO")

#define MAX 210000


long long fac[MAX], mmi[MAX];
int used[MAX], temp[MAX];

int N;
const int MAX_VAL = MAX;

vector<int> G[MAX];
void factorial_mod(){
	 fac[0]=fac[1]=1;
	for(long long i = 2; i < MAX_VAL; fac[i]%=MOD,i++)
		fac[i] = fac[i-1]*(i%MOD);
}

long long power_mod(long long x, long long n){
	long long ans = 1;
	for(;n;n>>=1,x*=x,ans%=MOD,x%=MOD)
		if(n&1)ans*=x;
	return ans%MOD;
}

void exgcd(int a, int b, int &x, int &y){
	if(b == 0){
		x = 1;
		y = 0;
		return ;
	}
	exgcd(b,a%b,y,x);
	y -= a/b * x;
}

void modular_multiplicatibe_inverse(){
	int x, y;  
	exgcd(fac[MAX_VAL-1],MOD,x,y);
	mmi[MAX_VAL-1] = x; 
	for(long long i = MAX_VAL-2; i >= 0; mmi[i]%=MOD,i--)
		mmi[i] = mmi[i+1]*((i+1)%MOD);
}

long long combination(long long n, long long r){
	return fac[n]*(mmi[r]*mmi[n-r]%MOD)%MOD;
}

void dfs(int depth = 0, int num = 1){
	if(used[num]) return;
	used[num] = true;
	
	temp[depth]++;
	
	for(int i = 0; i < G[num].size(); i++){
		if(used[G[num][i]]) continue;
		dfs(depth+1, G[num][i]);
	}
}
signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	
	int u, v;
	int ans = 0;
	
	cin>>N;
	
	factorial_mod();
	modular_multiplicatibe_inverse();
	
	for(int i = 0; i < N-1; i++){
		cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	
	dfs();
	
	for(int i = 0; i < N; i++){
		// cout<<i<<" "<<temp[i]<<" "<<ans<<endl;
		if(!temp[i]) break;
		ans += temp[i]*combination(N,i+1)%MOD*fac[i]%MOD*fac[N-i-1]%MOD;
		ans %= MOD;
		// cout<<combination(N,i+1)*fac[i]%MOD*fac[N-i-1]<<endl;;
	}
	
	cout<<ans<<endl;

	return 0;
}
0