【题解】洛谷P1850 换教室(暴力80)

直接顺序枚举换几节课，并枚举所有的情况并算出答案

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
using namespace std;
const int maxn=2100;
const int maxv=310;
const int maxe=90010;
int n,m,v,e;
int niu[maxn],gebi[maxn];
double k[maxn];
int to[maxv][maxv];
int tot;
double ans;
double minn;
double rminn=1e9;
double anss=1e9;
void floyd()
{
	for(int k=1;k<=v;k++)
	{
		for(int i=1;i<=v;i++)
		{
			for(int j=1;j<=v;j++)
			{
				if(i!=j&&j!=k&&i!=k)
				if(to[i][k]+to[k][j]<to[i][j])
				{
					to[i][j]=to[i][k]+to[k][j];
				}
			}
		}
	}
}
void dfs(int now,double a,double b,double ka,double kb,int mm)
{
	if(now==n)
	{
		anss=min(anss,a*ka+b*kb);
		return;
	}
	dfs(now+1,1,0,a*(ka+to[niu[now]][niu[now+1]])+b*(kb+to[gebi[now]][niu[now+1]]),0,mm);
	if(mm<m)
	{
		dfs(now+1,1-k[now+1],k[now+1],a*(ka+to[niu[now]][niu[now+1]])+b*(kb+to[gebi[now]][niu[now+1]]),a*(ka+to[niu[now]][gebi[now+1]])+b*(kb+to[gebi[now]][gebi[now+1]]),mm+1);
	}
}
int main()
{
//	freopen("classroom.in","r",stdin);
//	freopen("classroom.out","w",stdout);
	scanf("%d%d%d%d",&n,&m,&v,&e);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&niu[i]);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&gebi[i]);
	}
	for(int i=1;i<=n;i++)
	{
		cin>>k[i];
	}
	for(int i=1;i<=v;i++)
	{
		for(int j=1;j<=v;j++)
		{
			if(i==j) to[i][j]=0;
			else to[i][j]=1e9;
		}
	}
	for(int i=1;i<=e;i++)
	{
		int x,y,z;
		cin>>x>>y>>z;
		if(x==y) continue;
		to[x][y]=min(to[x][y],z);
		to[y][x]=min(to[y][x],z);	
		
	}
	floyd();
	if(m==0)
	{
		for(int i=1;i<=n-1;i++)
		{
			ans+=to[niu[i]][niu[i+1]]*1.0;
		}
		printf("%.2lf\n",ans);
		return 0;
	}
	if(m==1)
	{
		int tot=1;
		for(int i=1;i<=n-1;i++)
		{
			minn+=to[niu[i]][niu[i+1]]*1.0;
		}
		while(tot!=n)
		{
			ans=minn*(1.0-k[tot]);
			int md=0;
			for(int i=1;i<=n-1;i++)
			{	
				if(i==tot)
				{
					md+=to[gebi[i]][niu[i+1]];
				}
				if(i==tot-1)
				{
					md+=to[niu[i]][gebi[i+1]];
				}
				if(i!=tot&&i!=tot-1)
				{
					md+=to[niu[i]][niu[i+1]];
				}
			}
			ans+=md*k[tot];
			rminn=min(rminn,ans);
			tot++;
		}
		ans=minn*(1.0-k[n]);
		int md=0;
		for(int i=1;i<=n-1;i++)
		{	
			if(i==n-1)
			{
				md+=to[niu[i]][gebi[i+1]];
			}
			else
			{
				md+=to[niu[i]][niu[i+1]];
			}
		}
		ans+=md*k[n];
		rminn=min(rminn,ans);
		printf("%.2lf\n",rminn);
		return 0;
	}
	else
	{
		dfs(1,1,0,0,0,0);
		dfs(1,1-k[1],k[1],0,0,1);
		printf("%.2lf",anss);
	}
	fclose(stdin);
	fclose(stdout);
	return 0;
}