【HDU 5001】Walk（矩阵快速幂+概率DP）

这道题因为n不会太大，因此设dp[i]的意义为结尾的点为i的且满足条件的概率

因此可以将其写为一个1*n的矩阵，然后将其状态转移矩阵表示出来

将状态转移矩阵列出来了之后，将删除的点的转移全部改为0，这样就可以愉快地利用矩阵快速幂进行求解了

最后将dp[1..n]加起来就是该点的答案，时间复杂度n^4logd

#include <iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int SIZEN = 55;
int n;
vector<int> G[SIZEN];
struct Mat{
    double mat[SIZEN][SIZEN];
    void init(){
        memset(mat,0,sizeof(mat));
    }
    void getE(){
        memset(mat,0,sizeof(mat));
        for(int i = 0 ; i < n ; i++)
            mat[i][i] = 1;
    }
    Mat operator * (const Mat &a)const{
        Mat ret;
        ret.init();
        for(int i = 0 ; i < n ; i++)
            for(int j = 0 ; j < n ;j ++)
                for(int k = 0; k < n ;k ++)
                ret.mat[i][k] += mat[i][j] * a.mat[j][k];
        return ret;
    }
    void output(){
        for(int i = 0 ; i < n; i++){
            for(int j = 0; j < n ;j++) printf("%.5f ",mat[i][j]);
            printf("\n");
        }
    }
};
Mat operator ^(Mat a, int b){
    Mat ret;
    ret.getE();
    while(b){
        if(b & 1) ret = ret * a;
        a = a * a;
        b >>= 1;
    }
    return ret;
}
int in[SIZEN];
Mat C,t_C;
void init(){
    for(int i = 0 ; i < n; i++)
        G[i].clear();
    memset(in,0,sizeof(in));
}
void solve(){
    int m,d,u,v;
    init();
    scanf("%d%d%d",&n,&m,&d);
    for(int i = 0 ; i < m; i++){
        scanf("%d%d",&u,&v);
        u --; v--;
        in[u] ++;
        in[v] ++;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    C.init();
    for(int i = 0 ; i < n; i++){
        for(int j = 0; j < in[i];j++){
            C.mat[i][ G[i][j] ] = 1.0/in[i];
        }
    }
    for(int i = 0 ; i < n; i++){
        t_C = C;
        for(int j = 0; j < n; j++)
            t_C.mat[i][j] = t_C.mat[j][i] = 0;
        Mat ret;
        ret.init();
        for(int j = 0 ; j < n; j++) ret.mat[0][j] = 1.0/n;
        ret.mat[0][i] = 0;
        ret = ret * (t_C ^d);
        double ans = 0;
        for(int j = 0 ; j < n; j++) ans +=ret.mat[0][j];
        printf("%.10f\n",ans);
    }
}
int main()
{
    int _;
    scanf("%d",&_);
    while(_--) solve();
    return 0;
}