結果

問題 No.1418 Sum of Sum of Subtree Size
コンテスト
ユーザー tails
提出日時 2021-03-10 15:58:43
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 423 ms
コンパイル使用メモリ 32,384 KB
実行使用メモリ 8,320 KB
最終ジャッジ日時 2024-10-12 06:27:46
合計ジャッジ時間 2,909 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:28:1: warning: return type defaults to 'int' [-Wimplicit-int]
   28 | main(){
      | ^~~~
main.c: In function 'main':
main.c:58:9: warning: implicit declaration of function 'printf' [-Wimplicit-function-declaration]
   58 |         printf("%ld\n",a*2+n*n);
      |         ^~~~~~
main.c:1:1: note: include '<stdio.h>' or provide a declaration of 'printf'
  +++ |+#include <stdio.h>
    1 | #pragma GCC optimize("Ofast")
main.c:58:9: warning: incompatible implicit declaration of built-in function 'printf' [-Wbuiltin-declaration-mismatch]
   58 |         printf("%ld\n",a*2+n*n);
      |         ^~~~~~
main.c:58:9: note: include '<stdio.h>' or provide a declaration of 'printf'

ソースコード

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;}

long n,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 r=1;
	for(int k=0;k<en[i];++k){
		int j=eb[ei[i]+k];
		if(j!=p){
			long t=f(j,i);
			a+=t*(n-t);
			r+=t;
		}
	}
	return r;
}

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);
	printf("%ld\n",a*2+n*n);
}
0