【JZOJ5625】Max

最新推荐文章于 2023-05-04 14:29:39 发布

原创最新推荐文章于 2023-05-04 14:29:39 发布 · 270 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

题解同时被 2 个专栏收录

206 篇文章

订阅专栏

40 篇文章

订阅专栏

本文介绍了一种结合概率动态规划（DP）与状态压缩（状压）技巧的算法问题解决方法。通过定义状态转移方程，利用前缀和优化技巧来解决特定问题。文章详细解释了如何预处理转移矩阵，并通过代码示例展示了实现过程。

Description

这里写图片描述

Data Constraint

这里写图片描述

Solution

$m$ 较小，考虑对 $m$ 状压。设 $f_{i,s,j}$ 表示前 $i$ 个人，使用操作的状态为 $s$ ，最大值为 $j$ 的概率。
再设 $g_{i,s,j}$ 表示第 $i$ 个人，使用的操作状态 $s$ ，变成 $j$ 的概率。
预处理 $g$ ，求的时候前缀和优化一下就能过了。

Code

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;++i)
#define fd(i,j,k) for(int i=j;i>=k;--i)
#define ll long long
using namespace std;
const int N=45,M=11,mo=1e9+7;
int p[M][N][4];
int f[N][1<<10][32],g[N][1<<10][32];
int n,m,c;
void add(int &x,int y){
    x=x-mo+y;
    if(x<0) x+=mo;
}
int calc(int x){
    int t=0;
    fo(i,1,m) if((1<<i-1)&x) t++;
    return t;
}
int max(int x,int y){
    return x>y?x:y;
}
int main()
{
    freopen("max.in","r",stdin);
    freopen("max.out","w",stdout);
    scanf("%d %d %d",&n,&m,&c);
    fo(i,1,m)
    fo(j,1,n)
    fo(k,0,c) scanf("%d",&p[i][j][k]);
    fo(i,1,n){
        g[i][0][0]=1;
        fo(s,1,(1<<m)-1){
            int x=0,q=0;
            fo(j,1,m) if((1<<j-1)&s) {x=s^(1<<j-1),q=j;break;}
            fo(k,0,c)
            fo(j,0,m*c-k)
            add(g[i][s][j+k],(ll)g[i][x][j]*p[q][i][k]%mo);
        }
    }
    f[0][0][0]=1;
    fo(i,0,n-1){
        int o=(1<<m)-1;
        fo(s,0,o){
            int S=o^s;
            for(int x=S;x>=0;x=(x-1)&S){
                int sf=0,sg=0;
                fo(j,0,m*c){
                    add(sg,g[i+1][x][j]);
                    add(f[i+1][s^x][j],(ll)f[i][s][j]*sg%mo); 
                    add(f[i+1][s^x][j],(ll)g[i+1][x][j]*sf%mo);
                    add(sf,f[i][s][j]);
                }
                if(!x) break;
            }   
        }
    }
    int ans=0;
    fo(i,1,m*c) add(ans,(ll)f[n][(1<<m)-1][i]*i%mo);
    printf("%d",ans);
}