bzoj 2337: [HNOI2011]XOR和路径 期望dp+高斯消元

题意

给出一个无向连通图，边有边权。问从1走到n的期望异或值是多少（不一定是简单路径）。
n<=100,m<=10000,v<=10^9

分析

一开始想到了拆位后dp，但想到的是二维，第二维是0或1，算了一波复杂度发现不对，于是就没管了。

注意到每一位的贡献实际上是独立的，那么我们只要把每位的贡献分别计算就好了。
现在问题就变成了边权为0或1，问从1到n的异或值为1的概率。
设f[i]表示从i走到n异或值为1的概率，显然有$f[i]=\sum(1-f[to])*\frac{1}{d[i]}(边权为1)+\sum f[to]*\frac{1}{d[i]}(边权为0)$
边界条件为f[n]=0
高斯消元即可。

哈哈我用double居然过掉啦。

注意自环的话只要连一条边。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int N=105;
const double eps=1e-8;

int n,m,cnt,last[N],d[N];
double a[N][N];
struct edge{int to,next,val;}e[N*N*2];

void addedge(int u,int v,int val)
{
    e[++cnt].to=v;e[cnt].val=val;e[cnt].next=last[u];last[u]=cnt;
}

void build(int w)
{
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n+1;j++)
            a[i][j]=0;
    for (int i=1;i<n;i++)
    {
        a[i][i]=1;
        for (int j=last[i];j;j=e[j].next)
            if ((e[j].val&w)) a[i][n+1]+=(double)1.0/d[i],a[i][e[j].to]+=(double)1.0/d[i];
            else a[i][e[j].to]-=(double)1.0/d[i];
    }
    a[n][n]=1;
}

void gauss()
{
    for (int i=1;i<=n;i++)
    {
        int l=i;
        for (int j=i+1;j<=n;j++)
            if (fabs(a[j][i])>fabs(a[l][i])) l=j;
        if (l!=i) for (int j=1;j<=n+1;j++) swap(a[i][j],a[l][j]);
        double tmp=a[i][i];
        for (int j=1;j<=n+1;j++) a[i][j]/=tmp;
        for (int j=1;j<=n;j++)
            if (fabs(a[j][i])>eps&&i!=j)
            {
                tmp=a[j][i];
                for (int k=1;k<=n+1;k++) a[j][k]-=a[i][k]*tmp;
            }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    int mx=0;
    for (int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        mx=max(mx,z);d[x]++;
        addedge(x,y,z);
        if (x!=y) addedge(y,x,z),d[y]++;
    }
    double ans=0;
    for (int i=1;i<=mx;i<<=1)
    {
        build(i);
        gauss();
        ans+=1.0*a[1][n+1]*i;
    }
    printf("%.3lf",ans);
    return 0;
}