【概率dp 高斯消元】bzoj3270: 博物馆
一类成环概率dp的操作模式
Description
Input
Output
Sample Input
1 2
0.5
0.5
Sample Output
HINT
对于100%的数据有 n <= 20,n-1 <= m <= n(n-1)/2
题目分析
记$f[i][j]$为两人分别在$i$和$j$的概率,$u,v$为分别与$i,j$相连的点,那么有$$f[i][j]= \begin{cases} \sum \frac{f[u][v]*(1-p[u])*(1-p[v]))}{deg[u]*deg[v]}& \text{(u!=i&&v!=j)}\\\sum \frac{f[u][v]*(1-p[u])*p[v])}{deg[u]*deg[v]}& \text{(u!=i&&v=j)}\\\sum \frac{f[u][v]*p[u]*(1-p[v]))}{deg[u]*deg[v]}& \text{(u=i&&v!=j)}\\\sum \frac{f[u][v]*(1-p[u]*(1-p[v]))}{deg[u]*deg[v]}& \text{(u=i&&v=j)}\end{cases}$$
由于这里dp的关系成环,不能够直接转移,所以要使用高斯消元。由此共有$n^2$个状态,即$n^2$个未知量。故用$id[i][j]$来表示$f[i][j]$,就可方便地把这些未知量的关系用矩阵表示出来。
需要注意的是,初始$f[S][T]$的概率为1,不过后面又会有经过它的概率,所以在计算时需要考虑其初始概率。这里算是一个概率dp的奇怪的地方:最终$f[S][T]>1$。
再者是处理的一个小技巧:添一条自边便于处理。
#include<bits/stdc++.h>
const int maxn = ; int n,m,tot,S,T,deg[maxn],id[maxn][maxn];
double out[maxn],p[maxn],ans[maxn],mp[maxn][maxn];
int G[maxn][maxn]; void Gauss(int n)
{
double bse;
for (int i=, r; i<=n; i++)
{
r = i;
for (int j=i+; j<=n; j++)
if (fabs(mp[j][i]) > fabs(mp[r][i])) r = j;
if (r!=i) std::swap(mp[i], mp[r]);
bse = mp[i][i];
for (int j=i; j<=n+; j++) mp[i][j] /= bse;
for (int j=i+; j<=n; j++)
{
bse = mp[j][i];
for (int k=i; k<=n+; k++)
mp[j][k] -= bse*mp[i][k];
}
}
ans[n] = mp[n][n+];
for (int i=n-; i; i--)
{
ans[i] = mp[i][n+];
for (int j=i+; j<=n; j++) ans[i] -= ans[j]*mp[i][j];
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&S,&T);
tot = n*n;
for (int i=,x,y; i<=m; i++)
{
scanf("%d%d",&x,&y), ++deg[x], ++deg[y];
G[x][++G[x][]] = y, G[y][++G[y][]] = x;
}
for (int i=; i<=n; i++)
{
scanf("%lf",&p[i]);
G[i][++G[i][]] = i;
out[i] = (1.0-p[i])/(1.0*deg[i]);
}
for (int i=, t=; i<=n; i++)
for (int j=; j<=n; j++)
id[i][j] = ++t;
mp[id[S][T]][tot+] = -;
for (int i=; i<=n; i++)
for (int j=; j<=n; j++)
{
--mp[id[i][j]][id[i][j]];
for (int l=; l<=G[i][]; l++)
for (int r=; r<=G[j][]; r++)
{
int u = G[i][l], v = G[j][r];
if (u==v) continue;
if (u==i&&v==j) mp[id[i][j]][id[i][j]] += p[i]*p[j];
if (u!=i&&v!=j) mp[id[i][j]][id[u][v]] += out[u]*out[v];
if (u==i&&v!=j) mp[id[i][j]][id[u][v]] += p[u]*out[v];
if (u!=i&&v==j) mp[id[i][j]][id[u][v]] += out[u]*p[v];
}
}
Gauss(tot);
for (int i=; i<=n; i++) printf("%.6lf ",ans[id[i][i]]);
return ;
}
看到有网上有些博客说,高斯消元这里$mp[id[i][j]][id[u][v]]$是指$id[i][j]$状态转移到$id[u][v]$状态的概率,这个说法其实是不对的。$id[i][j]$实际上不过是和常规高斯消元一样,代表处理第几条方程;至于$id[u][v]$则表示当前这条方程中的$f[u][v]$;所以$mp[id[i][j]][id[u][v]]$是指第$id[i][j]$条方程中,第$id[u][v]$个未知量的系数。
END
最新文章
- sh脚本异常:/bin/sh^M:bad interpreter: No such file or directory
- Allegro padstack
- SubversionEdge 变更绑定地址
- Core Data
- Oracle导入导出dmp文件
- Remove Nth Node From End of List [LeetCode]
- Spring IOC 三种注入方式
- combo参数配置_手册
- 针对wordpress的二次开发
- c#winform关闭窗口时触发的事件
- BNU OJ 51005 BQG&#39;s Quadrilateral Bricks
- server
- Python打开新世界的大门-入门篇1
- Redis_集群_主从模式_哨兵模式
- 【BZOJ2159】Crash的文明世界(第二类斯特林数,动态规划)
- WEBAPI 的简单示例
- EF数据库配置
- HBase Filter程序样例及Shell(图)
- 定制NSError
- ACM-尼姆博弈之取(m堆)石子游戏——hdu2176