NOIP2016 换教室 概率DP求期望

题目链接

https://www.luogu.com.cn/problem/P1850

题意

v点e边带权无向连通图，给出n个三元组(a,b,p)，第i个数对代表第i个时间段，默认要去a，但可以申请(开始前一次性全部申请完成，不允许中途更改申请)去b，且通过申请的概率为p。

最多申请m次，问期望路径的最小值是多少。

思路

概率DP，我们定义dp[i][j][0]为第i阶段，使用了j次机会，本次不选择申请的期望路径，dp[i][j][1]类似，但是是这次选择申请的路径。

我们考虑转移方程，求的是期望路径，期望是有可加性的，两个状态转移时，其实变数，也就是概率部分只决定了移动的边，那么两个状态A-B的转移大概类似为A的期望+∑边权*选择这条边的概率。

得出转移方程

dp[i][j][0]=min(
                    dp[i-1][j][0]+gp[a[i-1]][a[i]],
                    dp[i-1][j][1]+gp[a[i-1]][a[i]]*(1-p[i-1])
                    			 +gp[b[i-1]][a[i]]*p[i-1]
                );

dp[i][j][1]=min(
                        dp[i-1][j-1][0]+gp[a[i-1]][b[i]]*p[i]+gp[a[i-1]][a[i]]*(1-p[i]),
                        dp[i-1][j-1][1]+gp[a[i-1]][a[i]]*(1-p[i])*(1-p[i-1])
                                    +gp[a[i-1]][b[i]]*(1-p[i-1])*p[i]
                                    +gp[b[i-1]][a[i]]*(1-p[i])*p[i-1]
                                    +gp[b[i-1]][b[i]]*p[i-1]*p[i]

                    );

其中gp为最短路径

代码

#include<cstdio>
#include<iostream>
#include<iomanip>
#include<map>
#include<unordered_map>
#include<string>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib> 
#include<chrono>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
//#define int long long
//#define double long double
using namespace std;
	typedef long long ll;
	const int maxn=2005;
    const int maxe=505;
	const int inf=0x3f3f3f3f;
	int n,m,v,e;
    int a[maxn],b[maxn];
    int gp[maxe][maxe];
    double p[maxn];
    double dp[maxn][maxn][2];
	signed main(){
        IOS
		#ifndef ONLINE_JUDGE
		    freopen("IO\\in.txt","r",stdin);
		    freopen("IO\\out.txt","w",stdout);
        #endif
		int tn=1;
        cin>>n>>m>>v>>e;
        for(int i=1;i<=n;i++)   cin>>a[i];
        for(int i=1;i<=n;i++)   cin>>b[i];
        for(int i=1;i<=n;i++)   cin>>p[i];
        memset(gp,0x3f,sizeof gp);
        while(e--){
            int uu,vv,ww;
            cin>>uu>>vv>>ww;
            gp[uu][vv]=gp[vv][uu]=min(gp[uu][vv],ww);
        }
        for(int i=0;i<maxn;i++) gp[i][i]=0;
        for(int k=1;k<=v;k++)
            for(int i=1;i<=v;i++)
                for(int j=1;j<=v;j++)
                    gp[i][j]=min(gp[i][j],gp[i][k]+gp[k][j]);
        for(int i=0;i<maxn;i++)
            for(int j=0;j<maxn;j++)
                dp[i][j][0]=dp[i][j][1]=inf;
        dp[0][0][0]=0;
        dp[1][1][1]=0;
        dp[1][0][0]=0;
        if(m>n)m=n;
        for(int i=2;i<=n;i++){
            for(int j=0;j<=min(m,i);j++){
                dp[i][j][0]=min(
                    dp[i-1][j][0]+gp[a[i-1]][a[i]],
                    dp[i-1][j][1]+gp[a[i-1]][a[i]]*(1-p[i-1])+gp[b[i-1]][a[i]]*p[i-1]
                );
                if(j){
                    dp[i][j][1]=min(
                        dp[i-1][j-1][0]+gp[a[i-1]][b[i]]*p[i]+gp[a[i-1]][a[i]]*(1-p[i]),
                        dp[i-1][j-1][1]+gp[a[i-1]][a[i]]*(1-p[i])*(1-p[i-1])
                                    +gp[a[i-1]][b[i]]*(1-p[i-1])*p[i]
                                    +gp[b[i-1]][a[i]]*(1-p[i])*p[i-1]
                                    +gp[b[i-1]][b[i]]*p[i-1]*p[i]

                    );
                }
            }
        }
       // for(int i=1;i<=n;i++){
       //     for(int j=0;j<=m;j++){
       //         cout<<setw(20)<<dp[i][j][0];
       //         cout<<" ";
       //         cout<<setw(20)<<dp[i][j][1];
       //         cout<<'|';
       //     }
       //     cout<<endl;
       // }
        double ans=inf;
        for(int i=0;i<=m;i++)
            ans=min(ans,min(dp[n][i][0],dp[n][i][1]));
        cout<<fixed<<setprecision(2)<<ans<<endl;
	}