AGC 039F - Min Product Sum

Posted 2022-12-31 weiyanpeng

考虑枚举行列的 \(n+m\) 元组，表示该行 / 列的最小值。

这样的方案数可以容斥轻松算出。

发现这样本质不同的限制有 \(2(n+m)\) 种：

• 当前行 / 列 \(> t \to \ge t+1\)

• 当前行 / 列 \(\ge t\)

每个位置上的限制是行列上的较大值，贡献是较小值。

\(f[cur][i][j]\) 表示当前考虑到加入 \(\ge cur\) 的限制，目前已经确定了 \(i\) 行 \(j\) 列上面的值。

依次加入，计算贡献即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,m,K,mod;
inline int add(int a,int b)a+=b;return a>=mod?a-mod:a;
inline int sub(int a,int b)a-=b;return a<0?a+mod:a;
inline int mul(int a,int b)return 1ll*a*b%mod;
inline int qpow(int a,int b)int ret=1;for(;b;b>>=1,a=mul(a,a))if(b&1)ret=mul(ret,a);return ret;
/* math */
const int N = 110;
int bin[N][N];
int c=0,f[2][N][N],cst[N][N];
int fac[N], ifac[N], pw[N][N*N];
int main()

    cin >> n >> m >> K >> mod;
    for(int i=0;i<=K;i++)
        pw[i][0]=1;for(int j=1;j<=n*m;j++)pw[i][j]=mul(pw[i][j-1],i);
    
    fac[0]=ifac[0]=1;for(int i=1;i<=max(n,m);i++)fac[i] = mul(fac[i-1],i);
    ifac[max(n,m)]=qpow(fac[max(n,m)],mod-2);for(int j=max(n,m)-1;j;j--)ifac[j]=mul(ifac[j+1],j+1);
    f[c][0][0]=1;
    for(int D = 1;D<=K;D++)
        c^=1;memset(f[c],0,sizeof(f[c]));
        for(int j=0;j<=n;j++)
        for(int k=0;k<=m;k++)
            cst[j][k] = mul(mul(ifac[j],pw[D-1][j*(m-k)]), mul(pw[K-D+1][j*k], j%2?mod-1:1));
        
        for(int i=0;i<=n;i++)for(int j=0;j<=m;j++)
            for(int k=0;k<=i;k++)
                f[c][i][j]=add(f[c][i][j], mul(f[c^1][i-k][j], cst[k][j]));
            
        
        c^=1;memset(f[c],0,sizeof(f[c]));
        for(int j=0;j<=m;j++)
        for(int k=0;k<=n;k++)
            cst[j][k] = mul(mul(ifac[j],pw[D-1][j*(n-k)]), mul(pw[K-D+1][j*k], j%2?mod-1:1));
        
        for(int i=0;i<=n;i++)for(int j=0;j<=m;j++)
            for(int k=0;k<=j;k++)
                f[c][i][j]=add(f[c][i][j], mul(f[c^1][i][j-k], cst[k][i]));
            
        
        c^=1;memset(f[c],0,sizeof(f[c]));
        for(int j=0;j<=n;j++)
        for(int k=0;k<=m;k++)
            cst[j][k] = mul(mul(ifac[j],pw[D][j*(m-k)]), mul(pw[K-D+1][j*k], 1));
        
        for(int i=0;i<=n;i++)for(int j=0;j<=m;j++)
            for(int k=0;k<=i;k++)
                f[c][i][j]=add(f[c][i][j], mul(f[c^1][i-k][j], cst[k][j]));
            
        
        c^=1;memset(f[c],0,sizeof(f[c]));
        for(int j=0;j<=m;j++)
        for(int k=0;k<=n;k++)
            cst[j][k] = mul(mul(ifac[j],pw[D][j*(n-k)]), mul(pw[K-D+1][j*k], 1));
        
        for(int i=0;i<=n;i++)for(int j=0;j<=m;j++)
            for(int k=0;k<=j;k++)
                f[c][i][j]=add(f[c][i][j], mul(f[c^1][i][j-k], cst[k][i]));
            
        
    
    cout << mul(f[c][n][m], mul(fac[n],fac[m])) << endl;