[BZOJ3270][高斯消元][概率与期望]博物馆

最新推荐文章于 2020-09-26 14:03:36 发布

LowestJN

最新推荐文章于 2020-09-26 14:03:36 发布

阅读量615

点赞数

分类专栏：高斯消元概率与期望文章标签：高斯消元概率与期望

本文链接：https://blog.csdn.net/coldef/article/details/62422908

版权

概率与期望同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

高斯消元

3 篇文章 0 订阅

订阅专栏

令 $f(x,y)$ 为一个人在x一个人在y的概率（与f(y,x)不同）

有四种转移
$f(x,y)*p[x]*p[y]\rightarrow f(x,y)$

$f(x,y)*p[x]*(1-p[y])/d[y]\rightarrow f(x,v)$ ， $v$ 为与 $y$ 相连的房间

$f(x,y)*(1-p[x])*p[y]/d[x]\rightarrow f(u,y)$ ， $u$ 为与 $x$ 相连的房间

$f(x,y)*(1-p[x])*(1-p[y])/d[x]/d[y]\rightarrow f(u,v)$

可以列出方程组

⎧ ⎩ ⎨ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ a 11 f (x 1, y 1) + a 12 f (x 2, y 2) + . . . + a 1 n f (x n, y n) = f (u 1, v 1) a 21 f (x 1, y 1) + a 22 f (x 2, y 2) + . . . + a 2 n f (x n, y n) = f (u 2, v 2) . . . a 31 f (x 1, y 1) + a 32 f (x 2, y 2) + . . . + a 3 n f (x n, y n) = f (u 3, v 3)

$\left\{ \begin{array}{c} a_{11}f(x_1,y_1)+a_{12}f(x_2,y_2)+...+a_{1n}f(x_n,y_n)=f(u_1,v_1)\\ a_{21}f(x_1,y_1)+a_{22}f(x_2,y_2)+...+a_{2n}f(x_n,y_n)=f(u_2,v_2)\\ ...\\ a_{31}f(x_1,y_1)+a_{32}f(x_2,y_2)+...+a_{3n}f(x_n,y_n)=f(u_3,v_3)\\ \end{array} \right.$

用高斯消元解出答案

#include <cstdio>
#include <algorithm>
#include <vector>
#include <cmath>
#include <iostream>
#define eps 1e-7
#define g(a,b) ((a-1)*n+b)

using namespace std;

int n,m,a,b,cnt,u,v,t,w,tot;
int d[25];
double A[410][410];
double p[25];
vector<int> E[25];

inline void Insert(int u,int v){
    E[u].push_back(v); E[v].push_back(u);
    d[u]++; d[v]++;
}

inline void check(int x,int y){
    double *B=A[g(x,y)];
    B[g(x,y)]=-1;
    for(int i=0;i<(int)E[x].size();i++)
        for(int j=0;j<(int)E[y].size();j++){
            u=E[x][i]; v=E[y][j]; w=g(u,v);
            if(u!=v){
                if(u==x&&v==y) B[w]+=p[u]*p[v];
                else if(u==x) B[w]+=p[u]*(1-p[v])/d[v];
                else if(v==y) B[w]+=p[v]*(1-p[u])/d[u];
                else B[w]+=(1-p[u])*(1-p[v])/d[v]/d[u];
            }
        }
}

inline void gauss(){
    int now=1;
    for(int i=1,j,k;i<=tot;i++){
        for(j=now;j<=tot&&fabs(A[j][i])<eps;j++);
        if(j>tot) continue;
        if(j!=now) for(k=1;k<=tot+1;k++) swap(A[j][k],A[now][k]);
        double t=A[now][i];
        for(j=1;j<=tot+1;j++) A[now][j]/=t;
        for(j=1;j<=tot;j++)
            if(j!=now){
                t=A[j][i];
                for(k=1;k<=tot+1;k++) A[j][k]-=t*A[now][k];
            }
        now++;
    }
}

int main(){
    scanf("%d%d%d%d",&n,&m,&a,&b);
    tot=n*n;
    A[g(a,b)][tot+1]=-1;
    for(int i=1;i<=n;i++) E[i].push_back(i);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        Insert(u,v);
    }
    for(int i=1;i<=n;i++) scanf("%lf",&p[i]);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        check(i,j);
    gauss();
    for(int i=1;i<=n;i++) printf("%.6lf ",A[g(i,i)][tot+1]);
}