結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー tailstails
提出日時 2021-03-10 15:58:43
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 361 ms
コンパイル使用メモリ 33,152 KB
実行使用メモリ 8,296 KB
最終ジャッジ日時 2024-04-20 10:52:13
合計ジャッジ時間 2,710 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 9 ms
6,940 KB
testcase_04 AC 9 ms
6,940 KB
testcase_05 AC 9 ms
6,940 KB
testcase_06 AC 9 ms
6,940 KB
testcase_07 AC 9 ms
6,940 KB
testcase_08 AC 6 ms
6,944 KB
testcase_09 AC 3 ms
6,944 KB
testcase_10 AC 4 ms
6,940 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 5 ms
6,940 KB
testcase_13 AC 6 ms
6,940 KB
testcase_14 AC 6 ms
6,940 KB
testcase_15 AC 5 ms
6,944 KB
testcase_16 AC 1 ms
6,940 KB
testcase_17 AC 1 ms
6,940 KB
testcase_18 AC 8 ms
6,940 KB
testcase_19 AC 2 ms
6,940 KB
testcase_20 AC 1 ms
6,944 KB
testcase_21 AC 5 ms
6,940 KB
testcase_22 AC 4 ms
6,940 KB
testcase_23 AC 1 ms
6,940 KB
testcase_24 AC 1 ms
6,940 KB
testcase_25 AC 1 ms
6,940 KB
testcase_26 AC 1 ms
6,944 KB
testcase_27 AC 1 ms
6,940 KB
testcase_28 AC 1 ms
6,940 KB
testcase_29 AC 1 ms
6,940 KB
testcase_30 AC 1 ms
6,944 KB
testcase_31 AC 1 ms
6,944 KB
testcase_32 AC 1 ms
6,940 KB
testcase_33 AC 3 ms
6,940 KB
testcase_34 AC 7 ms
8,296 KB
testcase_35 AC 3 ms
6,940 KB
testcase_36 AC 2 ms
6,944 KB
testcase_37 AC 4 ms
6,944 KB
testcase_38 AC 5 ms
6,944 KB
testcase_39 AC 1 ms
6,944 KB
testcase_40 AC 1 ms
6,944 KB
testcase_41 AC 1 ms
6,940 KB
testcase_42 AC 1 ms
6,940 KB
testcase_43 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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