http://acm.hdu.edu.cn/showproblem.php?pid=5001
题意:给定一个n点,m条边的有向图,在这n个点中随机选取一个起点,然后从这个点开始走d步,每一步以相同的概率到达与当前点相邻的点。问,d步后,不经过每个点的概率
思路1:设dp[i][j]表示走i步后到达点j的概率。
要求不经过点x的概率,那么每一步中到达点x的情况是不合法的,我们可以在处理每一步时 将到达点x的情况都去掉,即将dp[i][x]置0。那么d步走完后,到达其余点的路径中都不会包含点x,即最终到达其余点的概率和就是不经过点x的概率
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<vector>
using namespace std;
#define ll long long
//#define int long long
#define ull unsigned long long
#define PII pair<int,int>
#define mid ((l + r)>>1)
#define chl (root<<1)
#define chr (root<<1|1)
#define lowbit(x) ( x&(-x) )
const int manx = 1e4 + 10;
const int manx2 = 4e7 + 10;
const int INF = 2e9;
const int mod = 1e4+7;
int t,n,m,d,s,e;
vector<int>vt[60];
double dp[manx][60];
double solve(int x)
{
for(int i=1;i<=d;i++)
for(int j=1;j<=n;j++)dp[i][j]=0;
for(int i=1;i<=n;i++)dp[0][i]=1.0/n;
double ans=0;
for(int i=0;i<d;i++){
dp[i][x]=0;
for(int j=1;j<=n;j++){
int len=vt[j].size();
for(int k=0;k<len;k++)
dp[i+1][vt[j][k]]+=dp[i][j]/len;
}
}
dp[d][x]=0;
for(int i=1;i<=n;i++)ans+=dp[d][i];
return ans;
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d%d%d",&n,&m,&d);
for(int i=1;i<=n;i++)vt[i].clear();
for(int i=1;i<=m;i++){
scanf("%d%d",&s,&e);
vt[s].push_back(e);
vt[e].push_back(s);
}
for(int i=1;i<=n;i++)
printf("%.10f\n",solve(i));
}
return 0;
}
矩阵快速幂优化:在上面递推的基础上,因为每一次从上一层到下一层的线性转移方程都相同,所以可以考虑构造矩阵进行优化(不能经过点x时,将构造矩阵中x对应列的数置0即可)
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<vector>
using namespace std;
#define ll long long
//#define int long long
#define ull unsigned long long
#define PII pair<int,int>
#define mid ((l + r)>>1)
#define chl (root<<1)
#define chr (root<<1|1)
#define lowbit(x) ( x&(-x) )
const int manx = 1e4 + 10;
const int manx2 = 4e7 + 10;
const int INF = 2e9;
const int mod = 1e4+7;
int t,n,m,d,s,e;
vector<int>vt[60];
double mp[60][60];
struct mat
{
double a[60][60];
mat(){
memset(a,0,sizeof a);
}
friend mat operator*(const mat &x,const mat &y){
mat ans;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
for(int k=1;k<=n;k++)
ans.a[i][j]+=x.a[i][k]*y.a[k][j];
return ans;
}
}init,base;
mat q_pow(mat a,int b)
{
mat ans;
for(int i=1;i<=n;i++)ans.a[i][i]=1;
while(b){
if(b&1)
ans=ans*a;
a=a*a;
b>>=1;
}
return ans;
}
double solve(int x)
{
double sum=0;
mat init1=init,base1=base,ans;
init1.a[1][x]=0;
for(int i=1;i<=n;i++)base1.a[i][x]=0;
ans=q_pow(base1,d);
ans=init1*ans;
for(int i=1;i<=n;i++)sum+=ans.a[1][i];
return sum;
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d%d%d",&n,&m,&d);
memset(mp,0,sizeof mp);
for(int i=1;i<=n;i++)vt[i].clear();
for(int i=1;i<=m;i++){
scanf("%d%d",&s,&e);
vt[s].push_back(e);
vt[e].push_back(s);
mp[s][e]=mp[e][s]=1;
}
for(int i=1;i<=n;i++)
init.a[1][i]=1.0/n;
for(int j=1;j<=n;j++)
for(int i=1;i<=n;i++){
base.a[i][j]=1.0*mp[i][j]/vt[i].size();
}
for(int i=1;i<=n;i++)
printf("%.10f\n",solve(i));
}
return 0;
}
思路2:算出会经过点x的概率,最后用1减
我们将d步中某一步在点x的状态用1表示,不在点x的状态用0表示,那么我们可以分别求出以下几类路径的概率(X表示任意状态,容易知道这几类路径包含了会经过点x的所有情况),然后计算总和:
1XXXX……XXX(第一步经过点x)
01XXX……XXX(第一步不经过点x,第二步经过点x)
001XX……XXX(第一、二步不经过点x,第三步经过点x)
……
//转移方式和思路1相同,只是计算方式不同
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<vector>
using namespace std;
#define ll long long
//#define int long long
#define ull unsigned long long
#define PII pair<int,int>
#define mid ((l + r)>>1)
#define chl (root<<1)
#define chr (root<<1|1)
#define lowbit(x) ( x&(-x) )
const int manx = 1e4 + 10;
const int manx2 = 4e7 + 10;
const int INF = 2e9;
const int mod = 1e4+7;
int t,n,m,d,s,e;
vector<int>vt[60];
double dp[manx][60];
double solve(int x)
{
for(int i=1;i<=d;i++)
for(int j=1;j<=n;j++)dp[i][j]=0;
for(int i=1;i<=n;i++)dp[0][i]=1.0/n;
double ans=0;
for(int i=0;i<d;i++){
ans+=dp[i][x];
dp[i][x]=0;
for(int j=1;j<=n;j++){
int len=vt[j].size();
for(int k=0;k<len;k++)
dp[i+1][vt[j][k]]+=dp[i][j]/len;
}
}
return ans+=dp[d][x];
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d%d%d",&n,&m,&d);
for(int i=1;i<=n;i++)vt[i].clear();
for(int i=1;i<=m;i++){
scanf("%d%d",&s,&e);
vt[s].push_back(e);
vt[e].push_back(s);
}
for(int i=1;i<=n;i++)
printf("%.10f\n",1-solve(i));
}
return 0;
}
例:POJ-3744-Scout YYF I(概率dp+矩阵快速幂)