HDU-5001-Walk(概率dp,矩阵快速幂)

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+矩阵快速幂)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值