洛谷 P1850 [NOIP2016 提高组] 换教室【期望DP】

题目链接
题意：

牛牛要上 $n$ 节课，第 $i$ 节课在第 $c_i$ 个教室，牛牛有 $m$ 次机会申请吧教室 $c_i$ 换成 $d_i$，每节课的申请有 $d_i$ 的概率被接受，教室一共有$v$ 个，给出 $e$ 条无向边连接两个教室，每条边有个正整数代价，求上完 $n$ 节课所需的最小代价

题解：

设$f(i,j,0/1)$表示到第 $i$ 节课已经用了 $j$ 次申请，且这节课不使用$/$使用
申请的最小权值，用 Floyd 预处理一下最短路

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
 
const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 3e2+10;
int n,m,v,e;
int g[N][N],c[2005],d[2005];
double f[2005][2005][2],p[2005];
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>m>>v>>e;
	for(int i=1;i<=n;i++)cin>>c[i];
	for(int i=1;i<=n;i++)cin>>d[i];
	for(int i=1;i<=n;i++)cin>>p[i];
	
	for(int i=1;i<=v;i++)for(int j=1;j<=v;j++) g[i][j]=(i==j?0:INF);
	for(int i=1;i<=e;i++){
		int U,V,w; cin>>U>>V>>w; 
		g[U][V]=g[V][U]=min(w,g[U][V]);
	}
	for(int k=1;k<=v;k++)
		for(int i=1;i<=v;i++)
			for(int j=1;j<=v;j++)
				g[i][j]=min(g[i][j],g[i][k]+g[k][j]);
	for(int i=1;i<=n;i++)
		for(int j=0;j<=min(i,m);j++)f[i][j][0]=f[i][j][1]=INF;
	f[1][0][0]=f[1][1][1]=0;
	for(int i=2;i<=n;i++){
		for(int j=0;j<=min(i,m);j++){
			f[i][j][0]=min(f[i-1][j][0]+g[c[i-1]][c[i]],f[i-1][j][1]+g[c[i-1]][c[i]]*(1-p[i-1])+g[d[i-1]][c[i]]*p[i-1]);
      		if(j!=0)
        		f[i][j][1]=min(f[i-1][j-1][0]+g[c[i-1]][d[i]]*p[i]+g[c[i-1]][c[i]]*(1-p[i]),f[i-1][j-1][1]+g[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+g[c[i-1]][d[i]]*(1-p[i-1])*p[i]+g[d[i-1]][c[i]]*(1-p[i])*p[i-1]+g[d[i-1]][d[i]]*p[i-1]*p[i]);
		}
	}
	double ans=INF;
	for(int i=0;i<=m;i++)ans=min(ans,min(f[n][i][0],f[n][i][1]));
	cout<<fixed<<setprecision(2)<<ans<<endl;
}