結果

問題 No.1098 LCAs
コンテスト
ユーザー tails
提出日時 2020-11-04 17:55:57
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 1,067 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 31,488 KB
実行使用メモリ 25,344 KB
最終ジャッジ日時 2024-07-22 09:58:31
合計ジャッジ時間 2,898 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:38:1: warning: return type defaults to 'int' [-Wimplicit-int]
   38 | main(){
      | ^~~~
main.c: In function 'main':
main.c:72:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   72 |         write(1,wp,(wbuf+sizeof wbuf)-wp);
      |         ^~~~~
main.c:73:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   73 |         _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;}

char wbuf[20*200000];

#define MAXN 200000
int en[MAXN+1];
int ei[MAXN+1];
int eb[MAXN<<1];
union{
	int uv[MAXN<<1];
	struct{
		int ss1[MAXN+1];
		long ss2[MAXN+1];
	};
}u;
#define uv  u.uv
#define ss1 u.ss1
#define ss2 u.ss2

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

main(){
	char*rp=mmap(0l,7l*400001,1,2,0,0ll);
	RD(n);
	{
		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,0);
	char*wp=wbuf+sizeof wbuf;
	for(int i=n+1;--i;){
		long z=ss2[i];
		*--wp=10;
		while(*--wp=z%10+48,z/=10);
	}
	write(1,wp,(wbuf+sizeof wbuf)-wp);
	_exit(0);
}
0