洛谷 P2538 [SCOI2008]城堡（模拟退火）

传送门

---

解题思路

先用Floyd求一遍最短路，
然后对m个点之外的序列模拟退火，取前k个设置城堡，O(N^2)求出当前对小距离，判断并更新ans。
没什么注意的但是我rand()写在while外面了所以调了一个小时把我心态搞崩了草

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<ctime>
using namespace std;
const double delta=0.996;
int n,m,k,dis[55][55],r[55],vis[55],ans,cnt,a[55],x;
void floyd(){
	for(int k=1;k<=n;k++){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
			}
		}
	}
}
int cal(){
	if(k==cnt) return 0;
	int ans=0;
	for(int i=k+1;i<=cnt;i++){
		int res=1e9;
		for(int j=1;j<=n;j++){
			if(vis[j]){
				res=min(res,dis[a[i]][j]);
			}
		}
		ans=max(ans,res);
	}
	return ans;
}
void SA(){
	double t=3000;
	while(t>1e-15){
		int x=rand()%k+1,y=rand()%(cnt-k)+k+1;
		swap(a[x],a[y]);
		for(int i=1;i<=k;i++) vis[a[i]]=1;
		int res=cal();
		for(int i=1;i<=k;i++) vis[a[i]]=0;
		if(res<ans) ans=res;
		else if(exp((ans-res)/t)*RAND_MAX<rand()) swap(a[x],a[y]);
		t*=delta;
	}
}
int main(){
	ios::sync_with_stdio(false);
	srand(time(0));
	srand(rand());
	srand(rand());
	memset(dis,0x3f,sizeof(dis));
	cin>>n>>m>>k;
	if(m+k==n){
		cout<<0;
		return 0;
	}
	for(int i=1;i<=n;i++) cin>>r[i],r[i]++;
	for(int i=1;i<=n;i++) cin>>x,dis[i][r[i]]=dis[r[i]][i]=min(dis[i][r[i]],x);
	for(int i=1;i<=n;i++) dis[i][i]=0;
	floyd();
	for(int i=1;i<=m;i++) cin>>x,vis[x+1]=1;
	for(int i=1;i<=n;i++) if(!vis[i]) a[++cnt]=i;
	random_shuffle(a+1,a+cnt+1);
	ans=cal();
	while((double)clock()/CLOCKS_PER_SEC<=0.7) SA();
	cout<<ans;
	return 0;
}