分析
动态规划
先用Floyd预处理任意两点x,y之间的最短路d[x][y]
设f[ i ][ j ][ k ]表示当前是第i天,一共换了j间教室,k=0或1表示第i天是否换了教室
k[i]表示第i天换教室成功的概率
c[i][0]表示第i天换前的教室,c[i][1]表示第i天换后的教室
有以下四种情况
情况一:第i-1天换了教室,第i天未换教室
情况二:第i-1天换了教室,第i天换了教室
情况三:第i-1天未换教室,第i天未换教室
情况一:第i-1天未换教室,第i天换了教室
初始状态:
f[i][j][k]=inf(极大值)
f[1][0][0]=0; f[1][1][1]=0;
状态转移方程:
f[i][j][0]=min(f[i][j][0],f[i-1][j][1]+k[i-1]*d[c[i-1][1]][c[i][0]]+(1-k[i-1])*d[c[i-1][0]][c[i][0]]);
f[i][j][0]=min(f[i][j][0],f[i-1][j][0]+d[c[i-1][0]][c[i][0]]);
if(j!=0){
f[i][j][1]=min(f[i][j][1],f[i-1][j-1][0]+k[i]*d[c[i-1][0]][c[i][1]]+(1-k[i])*d[c[i-1][0]][c[i][0]]);
f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+k[i]*k[i-1]d[c[i-1][1]][c[i][1]]+k[i](1-k[i-1])*d[c[i-1][0]][c[i][1]]+(1-k[i])*k[i-1]d[c[i-1][1]][c[i][0]]+(1-k[i])(1-k[i-1])*d[c[i-1][0]][c[i][0]]);
}
代码如下
#include<bits/stdc++.h>
using namespace std;
const int N=2005;
const double maxi=800050000;
int n,m,v,e;
int c[N][2];
double k[N];
double d[N][N];
double f[N][N][2];
int read(){
int sum=0,f=1;
char ch=getchar();
while(ch>'9'||ch<'0')
{
if(ch=='-')f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
sum=(sum<<3)+(sum<<1)+ch-'0';
ch=getchar();
}
return sum*f;
}
int main(){
// freopen("classroom.in","r",stdin);
// freopen("classroom.out","w",stdout);
n=read();
m=read();
v=read();
e=read();
for(int i=1;i<=n;i++)
c[i][0]=read();
for(int i=1;i<=n;i++)
c[i][1]=read();
for(int i=1;i<=n;i++)
scanf("%lf",&k[i]);
for(int i=1;i<=v;i++)
for(int j=1;j<i;j++)
d[i][j]=d[j][i]=maxi;
int x,y;
double w;
for(int i=1;i<=e;i++)
{
x=read();
y=read();
scanf("%lf",&w);
if(w<d[x][y])d[x][y]=d[y][x]=w;
}
for(int t=1;t<=v;t++)
for(int i=1;i<=v;i++)
for(int j=1;j<i;j++)
if(d[i][j]>d[i][t]+d[t][j])
d[i][j]=d[j][i]=d[i][t]+d[t][j];
for(int i=1;i<=n;i++)
for(int j=0;j<=m;j++)
f[i][j][0]=f[i][j][1]=maxi;
f[1][0][0]=0;
f[1][1][1]=0;
for(int i=2;i<=n;i++)
for(int j=0;j<=m;j++)
{
f[i][j][0]=min(f[i][j][0],f[i-1][j][1]+k[i-1]*d[c[i-1][1]][c[i][0]]+(1-k[i-1])*d[c[i-1][0]][c[i][0]]);
f[i][j][0]=min(f[i][j][0],f[i-1][j][0]+d[c[i-1][0]][c[i][0]]);
if(j!=0){
f[i][j][1]=min(f[i][j][1],f[i-1][j-1][0]+k[i]*d[c[i-1][0]][c[i][1]]+(1-k[i])*d[c[i-1][0]][c[i][0]]);
f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+k[i]*k[i-1]*d[c[i-1][1]][c[i][1]]+k[i]*(1-k[i-1])*d[c[i-1][0]][c[i][1]]+(1-k[i])*k[i-1]*d[c[i-1][1]][c[i][0]]+(1-k[i])*(1-k[i-1])*d[c[i-1][0]][c[i][0]]);
}
}
double ans=maxi;
for(int j=0;j<=m;j++)
for(int t=0;t<2;t++)
{
ans=min(ans,f[n][j][t]);
}
printf("%.2lf",ans);
return 0;
}