結果

問題 No.1507 Road Blocked
コンテスト
ユーザー tails
提出日時 2021-05-16 22:32:15
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 1,305 bytes
コンパイル時間 453 ms
コンパイル使用メモリ 33,408 KB
実行使用メモリ 7,936 KB
最終ジャッジ日時 2024-10-05 09:43:37
合計ジャッジ時間 2,051 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:49:1: warning: return type defaults to 'int' [-Wimplicit-int]
   49 | main(){
      | ^~~~
main.c: In function 'main':
main.c:84:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   84 |         write(1,wbuf,wp-wbuf);
      |         ^~~~~
main.c:85:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   85 |         _exit(0);
      |         ^~~~~
      |         _Exit

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

char*mmap();
#define rd(v) int v=0;{int _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}

#define WTHI(v) {long _z=v,_n=0,_d=0;while(++_n,_d=_d<<8|0x30|_z%10,_z/=10);*(long*)wp=_d;wp+=_n;}
#define WTLO(v) {long _z=v,_n=8,_d=0;while(_d=_d<<8|0x30|_z%10,_z/=10,--_n);*(long*)wp=_d;wp+=8;}
#define wt(v) if(v>=100000000){WTHI(v/100000000);WTLO(v);}else{WTHI(v);}

#define MD 998244353

int inverse(int a){
	int b=MD;
	int u=1;
	int v=0;
	while(b){
		int q=a/b,t;
		t=b, b=a-q*b, a=t;
		t=v, v=u-q*v, u=t;
	}
	if(u<0){
		u+=MD;
	}
	return u;
}

long n;
long a;

#define MAXN 100000
int en[MAXN+1];
int ei[MAXN+1];
int eb[MAXN<<1];
int uv[MAXN<<1];

long f(int i,int p){
	long d=1;
	for(int k=0;k<en[i];++k){
		int j=eb[ei[i]+k];
		if(j!=p){
			d+=f(j,i);
		}
	}
	a+=d*(n-d);
	return d;
}

main(){
	char*rp=mmap(0l,1l<<28,1,2,0,0ll);

	{
		rd(t);
		n=t;
	}
	{
		int ne=n-1<<1;
		for(int j=0;j<ne;++j){
			rd(t);
			uv[j]=t;
			++en[uv[j]];
		}
	}
	{
		int s=0;
		for(int i=0;++i<=n;){
			ei[i]=s;
			s+=en[i];
			en[i]=0;
		}
	}
	{
		int ne=n-1<<1;
		for(int j=0;j<ne;++j){
			int i=uv[j];
			eb[ei[i]+en[i]++]=uv[j^1];
		}
	}
	f(1,1);
	long r=(MD+1-a%MD*inverse(n*(n-1)/2*(n-1)%MD)%MD)%MD;

	char wbuf[32],*wp=wbuf;
	wt(r);
	write(1,wbuf,wp-wbuf);
	_exit(0);
}
0