BZOJ 3143 游走 （高斯消元 贪心）

3143: [Hnoi2013]游走

Time Limit: 10 Sec Memory Limit: 128 MB
Description

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

Input

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

Output

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

Sample Input

3 3

2 3

1 2

1 3

Sample Output

3.333

HINT

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

思路：
摘自 < thy_asdf >
首先贪心是很显然的，为了使分数最小，期望到达次数越多的，编号就应该给的越小。
直接设边的期望列方程比较复杂。
所以我们换一个思路，先解出每个点的期望到达次数，那么边的期望次数就可以算出来了。
设边为y，与它相连的两个点为a，b，点的期望次数为f[a],f[b]，度数为du[a],du[b]；
那么边的期望次数g[y]就是f[a]/du[a]+f[b]/du[b]。
这很好理解，因为一条边只会从这两个点走过来。
现在考虑求点
对于一个点x，它只会从相邻的点过来，那么方程就是
f[x]=∑f[son[x]]/du[x];
除了n号点，我们就可以得到有n-1个未知数的n-1个方程了
高斯消元即可

#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 510
#define M 300010
using namespace std;

const double eps = 1e-10;/////////////////////////////

int n, m, idc;
int du[N];
double a[N][N], p[N], w[M];
double ans;

struct edge {
    int u, v;
}ed[M];

inline void adde(int x, int y) {
    ed[++idc].u = x, ed[idc].v = y;
    a[x][y] = a[y][x] = 1.0;
    ++du[x], ++du[y];
}

void gauss(int n) {
    int cnt = 0;
    for(int i=1; i<=n; i++) {
        int j = -1;
        for(int k=cnt+1; k<=n; k++) {
            if(fabs(a[k][i]) > eps) {
                j = k; break;
            }
        }
        if(j == -1) continue;
        for(int k=i; k<=n+1; k++) {
            swap(a[j][k], a[cnt+1][k]);
        }
        for(j=1; j<=n; j++) {
            if(j == cnt + 1) continue;
            if(fabs(a[j][i]) < eps) continue;
            double r = 0.0;
            if( a[cnt+1][i] ) r = a[j][i] / a[cnt+1][i];
            for(int k=i; k<=n+1; k++) {
                a[j][k] -= a[cnt+1][k] * r;
            }
        }
        cnt++;
    }
    for(int i=1; i<=n; i++)
        p[i] = a[i][n+1] / a[i][i];
}

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int main() {
    n = read(), m = read();
    for(int i=1; i<=m; ++i){
        int u = read(), v = read();
        adde(u, v);
    }
    for(int i=1; i<n; ++i) {
        for(int j=1; j<n; ++j)
            if( du[j] ) a[i][j] = a[i][j] / du[j];
        a[i][i] = -1, a[i][n] = 0;
    }
    a[1][n] = -1;
    gauss(n-1);
    for(int i=1; i<=n; ++i)
        if( du[i] ) p[i] /= du[i];
    for(int i=1; i<=m; ++i)
        w[i] = p[ed[i].u] + p[ed[i].v];
    sort(w+1, w+m+1);
    for(int i=1; i<=m; ++i) ans += w[i] * (m - i + 1);
    printf("%.3lf\n", ans);
    return 0;
}