题意
有n个点,第i个点到第j个点的路径在每个时刻有
p
i
,
j
p_{i,j}
pi,j的概率出现,每个时刻可以留在原地或走某条出现了的边。问在最优决策下从1走到n的期望时间是多少。
n
≤
1000
n\le1000
n≤1000
分析
设 f i f_i fi表示从 i i i开始走到 n n n的期望时间是多少。
考虑当我们位于某个点 x x x上时会如何决策,对于当前所有能到达的点,必然会选择到达一个 f y f_y fy最小的 y y y。
假设把所有点按 f f f值从小到大排序后得到的序列为 a 1 , a 2 , . . . , a n a_1,a_2,...,a_n a1,a2,...,an,显然有 a 1 = n a_1=n a1=n。不难得到转移 f a i = 1 + ∑ j = 1 i f a j ∗ p a i , a j ∏ k = 1 j − 1 ( 1 − p a i , a k ) f_{a_i}=1+\sum_{j=1}^if_{a_j}*p_{a_i,a_j}\prod_{k=1}^{j-1}(1-p_{a_i,a_k}) fai=1+j=1∑ifaj∗pai,ajk=1∏j−1(1−pai,ak)
若已经求出了 a 1 , ⋯ , a i a_1,\cdots,a_i a1,⋯,ai和 f a 1 , ⋯ , f a i f_{a_1},\cdots,f_{a_i} fa1,⋯,fai,把某个未选择的点 x x x放在 a i + 1 , ⋯ , a n a_{i+1},\cdots,a_n ai+1,⋯,an,得到的 f x f_x fx是单调不降的。故满足 a i + 1 = x a_{i+1}=x ai+1=x时 f x f_x fx最小的 x x x就是 a i + 1 a_{i+1} ai+1的实际值。
时间复杂度 O ( n 2 ) O(n^2) O(n2)。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
const int N=1005;
int n,p[N][N];
double f[N],g[N];
bool vis[N];
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
scanf("%d",&p[i][j]);
for (int i=1;i<n;i++) f[i]=g[i]=1;
for (int i=1;i<=n;i++)
{
double mn=1e9;int pos=0;
for (int j=1;j<=n;j++)
if (!vis[j]&&g[j]<1&&f[j]/(1-g[j])<mn) mn=f[j]/(1-g[j]),pos=j;
if (!pos) break;
vis[pos]=1;
f[pos]/=(1-g[pos]);
for (int j=1;j<=n;j++)
if (!vis[j])
{
f[j]+=g[j]*f[pos]*((double)p[j][pos]/100);
g[j]*=(double)(100-p[j][pos])/100;
}
}
printf("%.10lf",f[1]);
return 0;
}