結果

問題 No.1843 Tree ANDistance
ユーザー tails
提出日時 2022-03-08 18:23:13
言語 C
(gcc 13.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます
コンパイルメッセージ
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

ソースコード

diff # 

#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()));
}
0