luogu P3959 宝藏

题目描述

今有 $n$ 个点 $m$ 条边组成的无向图。选定一起点，则每条边的花费定义为$$L\times K$$
其中 $L$ 表示该边的权，$$K=\min (h[u],h[v])$$且 $h[u],h[v]$ 表示 $u,v$ 到起点的距离 $+1$。
求该图边花费和最小的生成树。

$\text{Solution P3959}$

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>

#define min(x,y) ((x)<(y)?(x):(y))
#define reg register

struct bide{
	int x,y,d,next;
}e[2010];
int len=0;
int n,m;
int first[20];
int sx,sy,sd;
int vis[20];
int ans=0x3f3f3f3f;
int sum;
bool tf[2010];
int h[20];
int cnt;

void ins(int x,int y,int d){
	e[++len].x=x;e[len].y=y;e[len].d=d;
	e[len].next=first[x];first[x]=len;
}
void work(int x){
	for(reg int i=first[x];i;i=e[i].next)
		tf[i]=1;
	sum=1;h[x]=1;cnt=0;vis[x]=1;
	while(sum!=n){
		int eg=rand()%len+1;
		if(!tf[eg]) continue;
		int y=e[eg].y;
		if(vis[y]){
			tf[eg]=0;
			continue;
		}
		vis[y]=1;
		h[y]=h[e[eg].x]+1;
		cnt+=e[eg].d*h[e[eg].x];
		for(reg int i=first[y];i;i=e[i].next)
			tf[i]=1;
		++sum;
	}
	ans=min(ans,cnt);
//	printf("%d\n",ans);
//	system("pause>nul");
}
int main(){
	srand(time(0));
	memset(first,0,sizeof(first));
	scanf("%d%d",&n,&m);
	for(reg int i=1;i<=m;++i){
		scanf("%d%d%d",&sx,&sy,&sd);
		ins(sx,sy,sd);ins(sy,sx,sd);
	}
	for(reg int t=1;t<=10000;++t)
	for(reg int i=1;i<=n;++i){
		memset(vis,0,sizeof(vis));
		memset(h,63,sizeof(h));
		memset(tf,0,sizeof(tf));
		work(i);
	}
	printf("%d",ans);
}