AtCoder Beginner Contest 277 G（概率dp+计数）

吃花椒的妙酱

已于 2022-11-28 10:28:08 修改

阅读量433

点赞数

文章标签：算法 c++ 数据结构

于 2022-11-28 10:17:45 首次发布

本文链接：https://blog.csdn.net/m0_53688600/article/details/128074201

版权

$\quad$ 题目大意：无向图有n个点m条边，人物初始level为0，初始在1号点，图上点有两种类型点0和1,遇到0会level++，遇到1会收益 += $level^2$ ，求k步内的收益期望(n<=3000,m<=3000)
$\quad$ Solution:
$\quad$ 一般期望题先考虑倒推，但是此题终态有点难定义（好像也有倒退做法）。考虑从定义角度入手。
$\quad$ 观察一种走法，设w为一种走法线路，f(w)表示走法的收益, $\sum_{i=1}^{len}X_i^2$ ，其中 $X_i$ 表示到i点的等级,而 $X_i = \sum_{j=1}^i[c_j=0],所以X_i^2=\sum_{j=1}^i\sum_{k=j}^i[c_j=0 \quad and \quad c_k=0]$ ,c表示点的类型。期望的计算 $E(w)=f(w)*\prod_{u=1}^{len-1} \frac{1}{deg(u)} = \sum_{u=1}^{len}X_u^2 * \prod_{v=1}^{u-1}\frac{1}{deg(v)} = \sum_{u=1}^{len} \prod_{v=1}^{u-1}\frac{1}{deg(v)} *\sum_{j=1}^u \sum_{k=j}^u[c_j==0 \&\&c_k=0]$ ,其中deg(u)表示u点出度,这样一来就转化成计数问题了。
$\quad$ 可以用dp来计数, $dp_{i,j,0/1,0/1}$ 表示走j步到了i点，上面那个柿子，点对类型为(0/1,0/1)的求和。
$\quad$ 转移的话，这里贴个代码吧，感觉更清楚
复杂度 $O (n * m a x (n, m))$

_for(i,0,k-1){
        _for(j,1,n){
            for(int u:G[j]){
                for(int a=0 ;a<2 ;a++){
                    for(int b=0 ;b<2 ;b++){
                        for(int na=a; na<2 ;na++){
                            for(int nb=b ; nb<2 ;nb++){
                                //下一步c[u]=1的话,na,nb不能变
                                if( c[u] && (na!=a || nb!=b) ) continue;
                                f[i+1][u][na][nb] = (f[i+1][u][na][nb] + i_du[j]*f[i][j][a][b]%mod)%mod;
                            }
                        }   
                    }
                }
            }
        }
    }

下一个点类型为1的时候，转移有限制。
$\quad$ 答案就是 $\sum_{i=1}^n\sum_{j=1}^kdp_{i,j,1,1}$
（代码写的dp[j][i][1/0][1/0]，前两维反了一下，懒得改了

完整代码

int c[N],i_du[N];
std::vector<int> G[N];
int f[N][N][2][2];
//第i步，到j的概率和，[0/1][0/1]，表示点对(x,y)
ll qsm(int a,int b){
    ll ans = 1 , tmp = a;
    while( b ){
        if( b&1 ) ans = ans * tmp%mod;
        tmp = tmp * tmp%mod;
        b>>=1;
    }
    return ans;
}
signed main(){ 
    IOS;
    int n,m,k;cin>>n>>m>>k;
    _for(i,1,m){
        int u,v;cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    _for(i,1,n) {
        cin>>c[i];
        i_du[i] = qsm((int)G[i].size(),mod-2);
    }
    //起点在1
    f[0][1][0][0] = 1;
    //枚举前i步
    _for(i,0,k-1){
        _for(j,1,n){
            for(int u:G[j]){
                for(int a=0 ;a<2 ;a++){
                    for(int b=0 ;b<2 ;b++){
                        for(int na=a; na<2 ;na++){
                            for(int nb=b ; nb<2 ;nb++){
                                //下一步c[u]=1的话,na,nb不能变
                                if( c[u] && (na!=a || nb!=b) ) continue;
                                f[i+1][u][na][nb] = (f[i+1][u][na][nb] + i_du[j]*f[i][j][a][b]%mod)%mod;
                            }
                        }   
                    }
                }
            }
        }
    }
    int ans = 0;
    _for(i,1,k){
        _for(j,1,n) if(c[j]) ans = (ans + f[i][j][1][1])%mod;
    }
    cout<<ans<<endl;
}