BZOJ 2337 [HNOI2011]XOR和路径 ——期望DP

首先可以各位分开求和

定义$f(i)$表示从i到n的期望值,然后经过一些常识,发现$f(n)=1$的时候的转移,然后直接转移,也可以找到$f(n)=0$的转移。

然后高斯消元31次就可以了。

#include <map>
#include <ctime>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
#define maxn 20005 int h[maxn],to[maxn],ne[maxn],du[maxn],w[maxn],en=0,n,m;
double a[105][105],ans[105]; void add(int a,int b,int c)
{du[a]++;to[en]=b;ne[en]=h[a];w[en]=c;h[a]=en++;} void Gauss()
{
F(i,1,n-1)
{
int tmp=i;
F(j,i,n-1)if(fabs(a[j][i])>fabs(a[tmp][i]))tmp=j;
F(j,i,n+1) swap(a[i][j],a[tmp][j]);
F(j,i+1,n-1)
{
double tmp=a[j][i]/a[i][i];
F(k,i,n+1) a[j][k]-=a[i][k]*tmp;
}
}
D(i,n-1,1)
{
F(j,i+1,n)
a[i][n+1]-=a[i][j]*ans[j];
ans[i]=a[i][n+1]/a[i][i];
}
} double ret=0.0; void solve(int x)
{
memset(a,0,sizeof a);
memset(ans,0,sizeof ans);
F(i,1,n-1)
{
for (int j=h[i];j>=0;j=ne[j])
if ((w[j]>>x)&1)
a[i][to[j]]+=1,a[i][n+1]+=1;
else a[i][to[j]]-=1;
a[i][i]+=du[i];
}
// F(i,1,n) F(j,1,n+1) printf("%.6f%c",a[i][j],j==n+1?'\n':' ');printf("\n\n");
Gauss();
ret+=ans[1]*(1<<x);
} int main()
{
memset(h,-1,sizeof h);
scanf("%d%d",&n,&m);
F(i,1,m)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);if (a!=b) add(b,a,c);
}
D(i,30,0) solve(i);
printf("%.3f\n",ret);
}

  

上一篇:bzoj 2337: [HNOI2011]XOR和路径


下一篇:【BZOJ 2337】 2337: [HNOI2011]XOR和路径(概率DP、高斯消元)