結果

問題 No.827 総神童数
ユーザー ok
提出日時 2019-05-03 23:30:48
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
AC  
実行時間 167 ms / 2,000 ms
コード長 1,914 bytes
コンパイル時間 830 ms
コンパイル使用メモリ 75,832 KB
実行使用メモリ 27,860 KB
最終ジャッジ日時 2024-12-31 19:15:46
合計ジャッジ時間 4,649 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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

#define MAX_VAL MAX

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

int ans, N;
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 permutation(long long n, long long r){
	if(n < r) return 0;
	return fac[n]*mmi[n-r]%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;
	
	ans += combination(N,depth+1)*fac[depth]%MOD*fac[N-depth-1]%MOD;
	
	// cout<<depth<<" "<<combination(N,depth+1)*fac[depth]<<endl;
	ans %= MOD;
	
	for(int i = 0; i < G[num].size(); i++){
		if(used[G[num][i]]) continue;
		// cout<<num<<" i = "<<i<<endl;
		dfs(depth+1, G[num][i]);
	}
}
signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	
	int u, v;
	
	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();
	
	cout<<ans<<endl;

	return 0;
}
0