Submission #1690830


Source Code Expand

#include <cmath>
#include <cstdio>
#include <algorithm>
#define Pi acos(-1.0)

using namespace std;

const int N=8e5+10,Mod=924844033;

struct Edge{ int to,next;} way[N<<1];

int n,num[N],tot,A[N],B[N],P[N],F[N<<4],fact[N],fact_inv[N];

void Build(int a,int b) { way[++tot]=(Edge){b,num[a]}; num[a]=tot; }

void Init()
{
	scanf("%d",&n);
	int a,b;
	for (int i=2;i<=n;++i)
	{
		scanf("%d%d",&a,&b);
		Build(a,b); Build(b,a);
	}
}

int son[N];
void Dfs(int x,int fa)
{
	son[x]=1;
	for (int i=num[x];i;i=way[i].next)
	{
		int v=way[i].to;
		if (v==fa) continue;
		Dfs(v,x);
		son[x]+=son[v];
		A[son[v]]--;
	}
	if (n-son[x]) A[n-son[x]]--;
}

int R[N<<2],L;
int inv(int x)
{
	int t=Mod-2,ret=1;
	while (t)
	{
		if (t&1) ret=1ll*ret*x%Mod;
		x=1ll*x*x%Mod;
		t>>=1;
	}
	return ret;
}

int qpow(int x,int k)
{
	int ret=1;
	while (k)
	{
		if (k&1) ret=1ll*ret*x%Mod;
		x=1ll*x*x%Mod;
		k>>=1;
	}
	return ret;
}

void BRC(int *y,int len)
{
	for (int i=1,j=len/2;i<len-1;++i)
	{
		if (i<j) swap(y[i],y[j]);
		int k=len/2;
		while (j>=k)
		{
			j-=k;
			k>>=1;
		}
		if (j<k) j+=k;
	}
}

void ntt(int *y,int len,int f)
{
	BRC(y,len);
	for (int h=2;h<=len;h<<=1)
	{
		int wn=qpow(5,(Mod-1)/h);
		if (f==-1) wn=qpow(wn,Mod-2);
		for (int j=0;j<len;j+=h)
		{
			int w=1;
			for (int k=j;k<j+h/2;++k)
			{
				int u=y[k],t=1ll*w*y[k+h/2]%Mod;
				y[k]=(u+t)%Mod;
				y[k+h/2]=(u-t+Mod)%Mod;
				w=1ll*w*wn%Mod;
			}
		}
	}
    if(f==-1){
        int ny=qpow(len,Mod-2);
        for(int i=0;i<len;i++) 
			y[i]=1ll*y[i]*ny%Mod;
    }
}

void Solve()
{
	Dfs(1,0);A[n]=n; 
	fact[0]=1; for (int i=1;i<=n;++i) fact[i]=1ll*fact[i-1]*i%Mod;
	int ny=inv(fact[n]);fact_inv[n]=ny;
	for (int i=n-1;i;--i) fact_inv[i]=1ll*fact_inv[i+1]*(i+1)%Mod;
	fact_inv[0]=1;
	for (int i=0;i<=n;++i) B[i]=fact_inv[i];
	for (int i=0;i<=n;++i) A[i]=1ll*(A[i]+Mod)*fact[i]%Mod;
	for (int i=0;i<=n;++i) P[i]=A[n-i];
	
	int len; for (len=1;len<((n+1)<<1);++len);
	ntt(P,len,1); ntt(B,len,1); 
    for(int i=0;i<len;i++)F[i]=1ll*P[i]*B[i]%Mod;
//    for (int i=0;i<len;++i) printf("%d%c",F[i],i==len-1?'\n':' ');
    ntt(F,len,-1);
	for (int i=n-1;i>=0;--i) 
		printf("%lld\n",1ll*F[i]*fact_inv[n-i]%Mod);
}

int main()
{
	Init();
	Solve();
	return 0;
}

Submission Info

Submission Time
Task F - Many Easy Problems
User Mcallor
Language C++14 (GCC 5.4.1)
Score 0
Code Size 2341 Byte
Status TLE
Exec Time 5259 ms
Memory 32640 KB

Compile Error

./Main.cpp: In function ‘void Init()’:
./Main.cpp:18:16: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d",&n);
                ^
./Main.cpp:22:22: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d",&a,&b);
                      ^

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 1900
Status
AC × 2
TLE × 1
AC × 2
TLE × 46
Set Name Test Cases
Sample example0, example1, example2
All doublestar0, doublestar1, doublestar2, doublestar3, doublestar4, example0, example1, example2, line0, line1, line2, line3, line4, maxrand0, maxrand1, maxrand10, maxrand11, maxrand12, maxrand13, maxrand14, maxrand15, maxrand16, maxrand17, maxrand18, maxrand19, maxrand2, maxrand3, maxrand4, maxrand5, maxrand6, maxrand7, maxrand8, maxrand9, rand0, rand1, rand2, rand3, rand4, rand5, rand6, rand7, rand8, rand9, star0, star1, star2, star3, star4
Case Name Status Exec Time Memory
doublestar0 TLE 5255 ms 24704 KB
doublestar1 TLE 5255 ms 24704 KB
doublestar2 TLE 5255 ms 24704 KB
doublestar3 TLE 5255 ms 24704 KB
doublestar4 TLE 5255 ms 24704 KB
example0 AC 6 ms 18560 KB
example1 TLE 5255 ms 16512 KB
example2 AC 6 ms 18560 KB
line0 TLE 5256 ms 32640 KB
line1 TLE 5256 ms 32640 KB
line2 TLE 5256 ms 32640 KB
line3 TLE 5256 ms 32640 KB
line4 TLE 5256 ms 32640 KB
maxrand0 TLE 5255 ms 24704 KB
maxrand1 TLE 5255 ms 24704 KB
maxrand10 TLE 5255 ms 24704 KB
maxrand11 TLE 5255 ms 24704 KB
maxrand12 TLE 5255 ms 24704 KB
maxrand13 TLE 5255 ms 24704 KB
maxrand14 TLE 5255 ms 24704 KB
maxrand15 TLE 5255 ms 24704 KB
maxrand16 TLE 5255 ms 24704 KB
maxrand17 TLE 5255 ms 24704 KB
maxrand18 TLE 5255 ms 24704 KB
maxrand19 TLE 5255 ms 24704 KB
maxrand2 TLE 5255 ms 24704 KB
maxrand3 TLE 5255 ms 24704 KB
maxrand4 TLE 5255 ms 24704 KB
maxrand5 TLE 5255 ms 24704 KB
maxrand6 TLE 5255 ms 24704 KB
maxrand7 TLE 5255 ms 24704 KB
maxrand8 TLE 5255 ms 24704 KB
maxrand9 TLE 5255 ms 24704 KB
rand0 TLE 5255 ms 16512 KB
rand1 TLE 5255 ms 16512 KB
rand2 TLE 5255 ms 16512 KB
rand3 TLE 5255 ms 16512 KB
rand4 TLE 5255 ms 16512 KB
rand5 TLE 5255 ms 16512 KB
rand6 TLE 5255 ms 16512 KB
rand7 TLE 5255 ms 16512 KB
rand8 TLE 5255 ms 16512 KB
rand9 TLE 5255 ms 16512 KB
star0 TLE 5255 ms 24704 KB
star1 TLE 5255 ms 24704 KB
star2 TLE 5255 ms 24704 KB
star3 TLE 5255 ms 24704 KB
star4 TLE 5259 ms 24704 KB