洛谷 P3232 [HNOI2013]游走 概率dp+高斯消元

题目描述

一个无向连通图，顶点从1编号到N，边从1编号到M。 小Z在该图上进行随机游走，初始时小Z在1号顶点，每一步小Z以相等的概率随机选 择当前顶点的某条边，沿着这条边走到下一个顶点，获得等于这条边的编号的分数。当小Z 到达N号顶点时游走结束，总分为所有获得的分数之和。 现在，请你对这M条边进行编号，使得小Z获得的总分的期望值最小。

输入输出格式

输入格式：
第一行是正整数N和M，分别表示该图的顶点数 和边数，接下来M行每行是整数u，v(1<=u,v<=N)，表示顶点u与顶点v之间存在一条边。 输入保证30%的数据满足N<=10，100%的数据满足2<=N<=500且是一个无向简单连通图。

输出格式：
仅包含一个实数，表示最小的期望值，保留3位小数。

输入输出样例

输入样例#1：
3 3
2 3
1 2
1 3
输出样例#1：
3.333
说明

边(1,2)编号为1，边(1,3)编号2，边(2,3)编号为3。

分析：设f[i]为从0点到i点的期望，就可以列出方程，然后上高斯消元。注意n点可以不管，1号点要先往周围走一步然后再随机走。除了1点的方程右边为1外，其他点都为0

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>

const int maxn=507;

using namespace std;

int ls[maxn],num[maxn];
double a[maxn][maxn],b[maxn],f[maxn],val[maxn*maxn*2],ans;
int n,m,i,cnt,j,x,y;

struct node{
    int x,y,w,next;
}edge[maxn*maxn*2];

void add(int x,int y)
{
    edge[++cnt].x=x;
    edge[cnt].y=y;
    edge[cnt].next=ls[x];
    ls[x]=cnt;
    num[x]++;
}

void guass(int n)
{
    int i,j,k;
    double t;
    for (i=1;i<=n;i++)
    {
        k=i;
        for (j=i+1;j<=n;j++)
         if (abs(a[j][i])>abs(a[k][i])) k=j;
        for (j=i+1;j<=n;j++) swap(a[i][j],a[k][j]);
        for (j=1;j<=n;j++) 
        {
            if (i==j) continue;
            double rate=a[j][i]/a[i][i];
            for (k=i;k<=n;k++) a[j][k]-=a[i][k]*rate;
            b[j]-=b[i]*rate;
        } 
    }   
    for (i=1;i<=n;i++) f[i]=b[i]/a[i][i];
}

int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }   
    for (x=1;x<n;x++)
    {
        a[x][x]=-1;
        for (i=ls[x];i>0;i=edge[i].next)
        {
            y=edge[i].y;
            if (y!=n)
            a[x][y]+=(double)1/num[y];
        }
    }   
    b[1]=-1;
    guass(n-1);
    for (i=1;i<=m;i++)
    {
        x=edge[i*2].x;
        y=edge[i*2].y;
        if (x!=n) val[i]+=f[x]/num[x];
        if (y!=n) val[i]+=f[y]/num[y];
    }
    sort(val+1,val+1+m);
    for (i=1;i<=m;i++)
    ans+=val[i]*(m-i+1);
    printf("%.3lf",ans);
}