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]\times P_i-a[i][j]\times P_j-a[i][k]\times P_k-\ldots = 0
\]

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

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

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

/*
12784kb 740ms
注意边数,没说就是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;
}
上一篇:专为物联网开发的开源操作系统Contiki(转)


下一篇:Spark API 之 combineByKey(一)