#744307 (C) No.1843 Tree ANDistance

• 問題: No.1843 Tree ANDistance
• ユーザー: tails
• 提出日時: 2022-03-08 18:23:13
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 42 ms / 2,000 ms
• コード長: 1,162 bytes
• コンパイル時間: 354 ms
• コンパイル使用メモリ: 31,232 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-07-23 18:26:00
• 合計ジャッジ時間: 4,023 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	41 ms; 6,812 KB
testcase_01	AC	41 ms; 6,816 KB
testcase_02	AC	37 ms; 6,944 KB
testcase_03	AC	37 ms; 6,940 KB
testcase_04	AC	41 ms; 6,940 KB
testcase_05	AC	30 ms; 6,944 KB
testcase_06	AC	42 ms; 6,940 KB
testcase_07	AC	39 ms; 6,940 KB
testcase_08	AC	40 ms; 6,940 KB
testcase_09	AC	36 ms; 6,940 KB
testcase_10	AC	35 ms; 6,940 KB
testcase_11	AC	41 ms; 6,940 KB
testcase_12	AC	26 ms; 6,940 KB
testcase_13	AC	37 ms; 6,940 KB
testcase_14	AC	2 ms; 6,940 KB
testcase_15	AC	2 ms; 6,940 KB
testcase_16	AC	1 ms; 6,940 KB
testcase_17	AC	1 ms; 6,944 KB
testcase_18	AC	1 ms; 6,940 KB
testcase_19	AC	2 ms; 6,944 KB
testcase_20	AC	2 ms; 6,940 KB
testcase_21	AC	1 ms; 6,940 KB
testcase_22	AC	0 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,944 KB
testcase_28	AC	7 ms; 6,944 KB
testcase_29	AC	5 ms; 6,944 KB
testcase_30	AC	4 ms; 6,940 KB
testcase_31	AC	8 ms; 6,940 KB
testcase_32	AC	8 ms; 6,940 KB
testcase_33	AC	3 ms; 6,944 KB
testcase_34	AC	6 ms; 6,940 KB
testcase_35	AC	1 ms; 6,940 KB
testcase_36	AC	1 ms; 6,944 KB
testcase_37	AC	1 ms; 6,940 KB

コンパイルメッセージ

main.c: In function 'f3':
main.c:63:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   63 |         write(1,wp,wbuf+sizeof wbuf-wp);
      |         ^~~~~
main.c:64:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   64 |         _exit(0);
      |         ^~~~~
      |         _Exit

ソースコード

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

#define rd(v) long v=0;{long _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}
#define wt(v) {ulong _z=v;do*--wp=_z%10+48;while(_z/=10);}
#define rep(v,e) for(long v=0;v<e;++v)

typedef unsigned long ulong;

#define UFSIZE 100001
int ufbuf[UFSIZE];

int uf_root(int a){
	int b;
	int t=a;
	while(b=ufbuf[t],b>=0) t=b;
	while(b=ufbuf[a],b>=0) ufbuf[a]=t,a=b;
	return t;
}

struct { int a,b; unsigned c; } abc[100000];

long f1(){
	char*mmap();
	char*rp=mmap(0l,1l<<25,1,2,0,0ll);
	rd(n);
	rep(i,n-1){
		{ rd(a); abc[i].a=a; }
		{ rd(b); abc[i].b=b; }
		{ rd(c); abc[i].c=c; }
	}
	return n;
}

long f2(long n){
	long z=0;
	for(unsigned m=1<<30;m;m>>=1){
		z<<=1;
		rep(i,n+1){
			ufbuf[i]=-1;
		}
		rep(i,n-1){
			if(abc[i].c&m){
				int a=uf_root(abc[i].a);
				int b=uf_root(abc[i].b);
				z+=ufbuf[a]*ufbuf[b];
				if(ufbuf[a]<ufbuf[b]){
					ufbuf[a]+=ufbuf[b];
					ufbuf[b]=a;
				}else{
					ufbuf[b]+=ufbuf[a];
					ufbuf[a]=b;
				}
			}
		}
	}
	return z%1000000007;
}

void f3(long z){
	char wbuf[64],*wp=wbuf+sizeof wbuf;
	wt(z);
	write(1,wp,wbuf+sizeof wbuf-wp);
	_exit(0);
}

int main(){
	f3(f2(f1()));
}