bzoj 3270: 博物馆 （高斯消元+概率）

3270: 博物馆

Time Limit: 30 Sec   Memory Limit: 128 MB
Submit: 356   Solved: 196
[ Submit][ Status][ Discuss]

Description

有一天Petya和他的朋友Vasya在进行他们众多旅行中的一次旅行，他们决定去参观一座城堡博物馆。这座博物馆有着特别的样式。它包含由m条走廊连接的n间房间，并且满足可以从任何一间房间到任何一间别的房间。

两个人在博物馆里逛了一会儿后两人决定分头行动，去看各自感兴趣的艺术品。他们约定在下午六点到一间房间会合。然而他们忘记了一件重要的事：他们并没有选好在哪儿碰面。等时间到六点，他们开始在博物馆里到处乱跑来找到对方（他们没法给对方打电话因为电话漫游费是很贵的）

不过，尽管他们到处乱跑，但他们还没有看完足够的艺术品，因此他们每个人采取如下的行动方法：每一分钟做决定往哪里走，有Pi 的概率在这分钟内不去其他地方（即呆在房间不动），有1-Pi 的概率他会在相邻的房间中等可能的选择一间并沿着走廊过去。这里的i指的是当期所在房间的序号。在古代建造是一件花费非常大的事，因此每条走廊会连接两个不同的房间，并且任意两个房间至多被一条走廊连接。

两个男孩同时行动。由于走廊很暗，两人不可能在走廊碰面，不过他们可以从走廊的两个方向通行。（此外，两个男孩可以同时地穿过同一条走廊却不会相遇）两个男孩按照上述方法行动直到他们碰面为止。更进一步地说，当两个人在某个时刻选择前往同一间房间，那么他们就会在那个房间相遇。

两个男孩现在分别处在a，b两个房间，求两人在每间房间相遇的概率。

Input

第一行包含四个整数，n表示房间的个数;m表示走廊的数目;a,b (1 ≤ a, b ≤ n),表示两个男孩的初始位置。

之后m行每行包含两个整数，表示走廊所连接的两个房间。

之后n行每行一个至多精确到小数点后四位的实数 表示待在每间房间的概率。

题目保证每个房间都可以由其他任何房间通过走廊走到。

Output

输出一行包含n个由空格分隔的数字，注意最后一个数字后也有空格，第i个数字代表两个人在第i间房间碰面的概率（输出保留6位小数）

注意最后一个数字后面也有一个空格

Sample Input

2 1 1 2
1 2
0.5
0.5

Sample Output

0.500000 0.500000

HINT

对于100%的数据有 n <= 20，n-1 <= m <= n(n-1)/2

Source

高斯消元

[ Submit][ Status][ Discuss]

题解：高斯消元+概率

设节点i的度为du[i]

那么从i到达与他相邻的任意节点的概率out[i]就是(1-p[i])/du[i]

设第一个人在x,第二个人在y的概率为E(x,y)

枚举与x相邻的点u，与y相邻的点v

E(x,y)=E(u,v)*out[u]*out[v]+E(x,y)*p[x]*p[y]+E(x,v)*p[x]*out[v]+E(u,y)*p[y]*out[u]

我们可以将(x,y)看成一个状态，那么我们对于每个状态都可以得到这么一个每个状态相互关联的式子。

然后移项可得0=E(u,v)*out[u]*out[v]+E(x,y)*p[x]*p[y]+E(x,v)*p[x]*out[v]+E(u,y)*p[y]*out[u]-E(x,y)

可以得到n*n个方程组，我们可以用高斯消元来求解每个E(x,y)

需要特别注意的是E(s,t)刚开始的概率是1，后来又会有再次经过的概率，但是我们列式子的时候要把左边的常数项赋值成-1.

#include<iostream>  
#include<cstdio>  
#include<algorithm>  
#include<cstring>  
#include<cmath>  
#define N 600 
#define id(x,y) n*(x-1)+y  
#define eps 1e-9  
using namespace std;  
int n,m,point[N],v1[N<<1],next[N<<1],du[N],s,t,tot,size;  
double f[N],a[N][N],p[N],out[N],ans[N];   
void add(int x,int y)  
{  
    tot++; next[tot]=point[x]; point[x]=tot; v1[tot]=y;  
    if (x!=y) {
	 tot++; next[tot]=point[y]; point[y]=tot; v1[tot]=x; }  
}  
void build(int x,int y)  
{  
    a[id(x,y)][id(x,y)]=-1; 
    for (int i=point[x];i;i=next[i])  
     for (int j=point[y];j;j=next[j]){  
        int u=v1[i]; int v=v1[j];  
        if (v!=u) {  
            if (u==x&&v==y) a[id(x,y)][id(u,v)]+=p[x]*p[y];  
            else if (u==x&&v!=y) a[id(x,y)][id(u,v)]+=p[x]*out[v];  
            else if (u!=x&&v==y) a[id(x,y)][id(u,v)]+=out[u]*p[y];  
            else if (u!=x&&v!=y) a[id(x,y)][id(u,v)]+=out[u]*out[v];  
         }  
     }  
}  
void gauss()  
{  
    for (int i=1;i<size;i++) {  
        int num=i;  
        for (int j=i+1;j<=size;j++)  
         if (fabs(a[j][i])>fabs(a[num][i])) num=j;  
        if (num!=i)   
         for (int j=1;j<=size+1;j++) swap(a[num][j],a[i][j]);  
        for (int j=i+1;j<=size;j++) {  
            if (fabs(a[j][i])<eps) continue;  
            double t=a[j][i]/a[i][i];  
            for (int k=1;k<=size+1;k++)  
             a[j][k]-=t*a[i][k];  
        }  
    }  
    for (int i=size;i;i--) {  
        double t=1.0*a[i][size+1];  
        for (int j=i+1;j<=size;j++) t-=1.0*ans[j]*a[i][j];  
        ans[i]=t/a[i][i];  
    }  
}  
int main()  
{  
    freopen("a.in","r",stdin);  
    //freopen("my.out","w",stdout);
    scanf("%d%d",&n,&m);  
    scanf("%d%d",&s,&t);  
    size=n*n;  
    a[id(s,t)][size+1]=-1;  
    for (int i=1;i<=m;i++) {  
        int x,y; scanf("%d%d",&x,&y);  
        add(x,y); du[x]++; du[y]++;  
    }  
    for (int i=1;i<=n;i++) scanf("%lf",&p[i]);  
    for (int i=1;i<=n;i++) add(i,i),out[i]=(1.0-p[i])/(double)du[i];
	//for (int i=1;i<=n;i++) cout<<out[i]<<" ";
	//cout<<endl;   
    for (int i=1;i<=n;i++)  
     for (int j=1;j<=n;j++) build(i,j); 
	//for (int i=1;i<=size+1;i++,cout<<endl)
	 //for (int j=1;j<=size+1;j++)
	 // cout<<a[i][j]<<" "; 
    gauss();  
    for (int i=1;i<=n;i++)   
        printf("%.6lf ",fabs(ans[id(i,i)]));  
    printf("\n");  
}