【题解】洛谷P3959 宝藏（生成树 随机化）

一开始可能大部分人的想法是这是个最小生成树问题，用prim就能解决，然而实际上可以发现由于有倍数与深度的限制，这种算法是不一定正确的。那该怎么办呢？我们只需要在找最小点之前做一下判断，用随机数取模，保证其有很小的概率取出来的是次小、更小的概率取出来的是次次小……的点，这个过程重复1000次（差不多吧），基本上就能通过了。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iomanip>
#include<queue>
#include<cmath>
#include<ctime>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=20;
const int maxm=1010;
int e[maxn][maxn];
int dis[maxn],depth[maxn];
bool b[maxn],inq[maxn];
struct H
{
	int id;
};
int n,m,ans;
priority_queue<H> q;
queue<H> p;
bool operator < (const H &a,const H &b)
{
	return dis[a.id]>dis[b.id];
}
int f(int x)
{
	memset(dis,INF,sizeof(dis));
	memset(b,false,sizeof(b));
	memset(inq,false,sizeof(inq));
	memset(depth,0,sizeof(depth));
	while(!q.empty()) q.pop();
	while(!p.empty()) p.pop();
	
	dis[x]=0;
	depth[x]=1;
	q.push((H){x});
	inq[x]=true;
	int tot=0;
	for(int i=1;i<=n;i++)
	{
		H now=q.top();
		q.pop();
		while(!q.empty()&&(rand()%10==0))
		{
			p.push(now);
			now=q.top();
			q.pop();
		}
		while(!p.empty()) 
		{
			q.push(p.front());
			p.pop();
		}
		int tmp=now.id;
		b[tmp]=true;
		tot+=dis[tmp];
		if(tot>ans) return INF;
		for(int j=1;j<=n;j++)
		{
			if(!b[j]&&e[tmp][j]!=INF&&dis[j]>depth[tmp]*e[tmp][j])
			{
				dis[j]=depth[tmp]*e[tmp][j];
				depth[j]=depth[tmp]+1;
				if(!inq[j])
				{
					inq[j]=true;
					q.push((H){j});	
				}			
			}
		}
	}
	return tot;
}
int main()
{
	scanf("%d %d",&n,&m);
	memset(e,INF,sizeof(e));
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		scanf("%d %d %d",&x,&y,&z);
		if(z<e[x][y])
		{
			e[x][y]=z;
			e[y][x]=z;
		}
	}
	ans=INF;
	srand(1919);
	
	for(int i=1;i<=1000;i++)
	{
		for(int j=1;j<=n;j++)
		{
			ans=min(ans,f(j));
		}
	}
	
	printf("%d",ans);
	return 0;
}