結果

問題 No.2200 Weird Shortest Path
ユーザー tails
提出日時 2023-01-28 23:59:03
言語 C90
(gcc 12.3.0)
結果
AC  
実行時間 12 ms / 2,000 ms
コード長 1,713 bytes
コンパイル時間 771 ms
コンパイル使用メモリ 26,240 KB
実行使用メモリ 8,916 KB
最終ジャッジ日時 2024-06-29 00:53:46
合計ジャッジ時間 2,857 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 44
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function ‘main’:
main.c:7:59: warning: implicit declaration of function ‘write’ [-Wimplicit-function-declaration]
    7 | #define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
      |                                                           ^~~~~
main.c:93:9: note: in expansion of macro ‘wt1’
   93 |         wt1(z);
      |         ^~~
main.c:94:9: warning: implicit declaration of function ‘_exit’ [-Wimplicit-function-declaration]
   94 |         _exit(0);
      |         ^~~~~
main.c:94:9: warning: incompatible implicit declaration of built-in function ‘_exit’ [-Wbuiltin-declaration-mismatch]

ソースコード

diff #

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

#define getrp() ({char*mmap();mmap(0l,1l<<25,1,2,0,0ll);})
#define rd() ({long _v=0,_c;while(_c=*rp++-48,_c>=0)_v=_v*10+_c;_v;})
#define wt(v) ({ulong _z=v;do*--wp=_z%10+48;while(_z/=10);})
#define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
#define rep(v,e) for(long v=0;v<e;++v)

typedef unsigned long ulong;

ulong e0[200000],e1[200000];

void radix_sort_aux(ulong*a,ulong*b,long n,long last){
	int c[512];
	for(long i=0;i<512;++i){
		c[i]=0;
	}
	for(long i=0;i<n;++i){
		++c[a[i]&511];
	}
	int t=0;
	for(long i=0;i<512;++i){
		int u=c[i];
		c[i]=t;
		t+=u;
	}
	if(!last){
		for(long i=0;i<n;++i){
			b[c[a[i]&511]++]=a[i]>>9|a[i]<<55;
		}
	}else{
		for(long i=0;i<n;++i){
			b[c[a[i]&511]++]=a[i]>>46|a[i]<<18;
		}
	}
}

#define UFSIZE 100001
int ufbuf[UFSIZE];
void uf_init(){
	for(int i=0;i<UFSIZE;++i){
		ufbuf[i]=-1;
	}
}
static inline
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;
}
static inline
void uf_unite(int a,int b){
	//a=uf_root(a);
	//b=uf_root(b);
	if(a!=b){
		if(ufbuf[a]<ufbuf[b]){
			ufbuf[a]+=ufbuf[b];
			ufbuf[b]=a;
		}else{
			ufbuf[b]+=ufbuf[a];
			ufbuf[a]=b;
		}
	}
}

int main(){
	char*rp=getrp();
	long n=rd();
	long m=rd();
	rep(i,m){
		ulong a=rd();
		ulong b=rd();
		ulong w=rd();
		e0[i]=w|a<<27|b<<44;
	}
	radix_sort_aux(e0,e1,m,0);
	radix_sort_aux(e1,e0,m,0);
	radix_sort_aux(e0,e1,m,1);
	uf_init();
	ulong z=0;
	rep(i,m){
		ulong e=e1[i];
		long a=uf_root(e>>27&0x1ffff);
		long b=uf_root(e>>44&0x1ffff);
		if(a!=b){
			z+=(e&0x7ffffff)*ufbuf[a]*ufbuf[b];
			uf_unite(a,b);
		}
	}
	wt1(z);
	_exit(0);
}
0