BZOJ.3143.[HNOI2013]游走(概率 期望 高斯消元)

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

题目链接

参考 远航之曲

把走每条边的概率乘上分配的标号就是它的期望，所以我们肯定是把大的编号分配给走的概率最低的边。

我们只要计算出经过所有点的概率，就可以得出经过一条边(\(u->v\))的概率\(P_{ei}\)。用\(dgr[i]\)表示点\(i\)的度数，那么$$P_{ei}=\frac{P_u}{dgr[u]}+\frac{P_v}{dgr[v]}$$

每个点的概率怎么求呢？就是$$P_i=\sum_{(i,j)\in G}\frac{P_j}{dgr[j]}$$

用\(a[i][j]\)为从点\(j\)走到点\(i\)的概率，即$$a[i][j]=\frac{1}{dgr[j]}$$

那么每个点的概率$$P_i=\sum_{(i,j)\in G} a[i][j]\times P[j]$$

注意走到\(n\)就不会再走了，所以高斯消元不处理与\(n\)有关的东西。

\(a[i][i]=1\)，然后把每个概率\(P_i\)看做一个变量，可以列出\(n-1\)个含\(n-1\)个未知数的方程

每个\(a[i][i]\)对应的变量就是\(P_i\)了。

因为一开始就在\(1\)号点，所以第一个方程的结果应设为\(1\)，即\(P_1=1+\sum
\frac{P_j}{dgr[j]}\)。

高斯消元，复杂度\(O(n^3)\)，好像有\(\frac{1}{6}\)的常数？

/*
12784kb740ms
注意边数，没说就是n^2级别的！
精度(eps)要高！
高斯写错。。
*/
#include <cmath>
#include <cstdio>
#include <cctype>
#include <iostream>
#include <algorithm>
#define gc() getchar()
#define eps 1e-12
const int N=505,M=N*N*2;int n,m,Enum,H[N],fr[M],to[M],nxt[M],dgr[N];
double A[N][N],Ans[N],pe[M];inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
inline void AddEdge(int u,int v)
{
++dgr[u], to[++Enum]=v, fr[Enum]=u, nxt[Enum]=H[u], H[u]=Enum;
++dgr[v], to[++Enum]=u, fr[Enum]=v, nxt[Enum]=H[v], H[v]=Enum;
}
inline bool cmp(const double &a,const double &b){
return std::fabs(a)>std::fabs(b);
}
void Gauss()
{
for(int j=1; j<n; ++j)
{
int mxrow=j;
for(int i=j+1; i<n; ++i)
if(cmp(A[i][j],A[mxrow][j])) mxrow=i;
if(mxrow!=j) std::swap(A[mxrow],A[j]);//和下一行也差不多
//if(mxrow!=j) for(int i=1; i<=n; ++i) std::swap(A[mxrow][i],A[j][i]);
for(int i=j+1; i<n; ++i)
if(fabs(A[i][j])>eps)//判一下显然快(吗?) 但要注意精度
//if(A[i][j])
{
double t=A[i][j]/A[j][j];
for(int k=j; k<=n; ++k)
A[i][k]-=A[j][k]*t;
}
}
for(int i=n-1; i; --i)
{
for(int j=i+1; j<n; ++j) A[i][n]-=A[i][j]*Ans[j];
Ans[i]=A[i][n]/A[i][i];
}
}int main()
{
n=read(),m=read();
for(int u,v,i=1; i<=m; ++i) u=read(),v=read(),AddEdge(u,v);for(int x=1; x<n; A[x][x]=1.0,++x)
for(int i=H[x]; i; i=nxt[i])
if(to[i]!=n) A[x][to[i]]=-1.0/dgr[to[i]];
A[1][n]=1.0;
Gauss();for(int i=1; i<=m; ++i) pe[i]=Ans[fr[i<<1]]/dgr[fr[i<<1]]+Ans[to[i<<1]]/dgr[to[i<<1]];
std::sort(pe+1,pe+1+m,std::greater<double>());
double res=0;
for(int i=1; i<=m; ++i) res+=1.0*i*pe[i];
printf("%.3lf",res);return 0;
}