给出一个矩阵,要求每行只能选一个节点,每列选的节点不能超过所有选的节点的一半,给出每个节点的选择方案数(a[i][j] )
不能完全不选,求总方案数
直接容斥,ALL - 存在一个列的所选个数超过一半
但是发现这种列只能存在一个(脑补
枚举这个列, 对这个列做dp : F [ i] [j ][ k ], 前i行 ,该列所选个数为j ,未选的个数为k, 的方案数
F[ i ] [ j ] [k] = F[i-1][j][k]
F[i][j][k] += F[ i - 1][ j-1 ] [ k ] *a[i][column]
+=F[i-1][ j ][k-1] * (s[i] -a[i][column] )
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std ;
const int N =102,M =2100;
const int mod = 998244353 ;
#define int long long
int A[M],B[M];
int f[N][N][N],n,m,a[N][M],s[N];
void solve(){
int i,j;
int ALL=1;
cin>>n>>m;
for(i=1;i<=n;i++){
s[i]=0;
for(j=1;j<=m;j++)
cin>>a[i][j],s[i]+=a[i][j],s[i]%=mod;
ALL*= (s[i]+1),ALL%=mod;
}
int answer =0;
for(int p=1;p<=m;p++){
memset(f,0,sizeof f);
for(i=1;i<=n;i++)
A[i]=a[i][p],B[i]=(s[i]-A[i]+mod)%mod;
f[0][0][0]=1;
for(i=1;i<=n;i++)
for(j=0;j<=i;j++)
for(int k=0;k<=i-j;k++){
f[i][j][k]=f[i-1][j][k];
if(j>=1) f[i][j][k]+=(f[i-1][j-1][k]*A[i]);
if(k>=1) f[i][j][k]+=(f[i-1][j][k-1]*B[i]);
f[i][j][k]%=mod;
}
for(i=1;i<=n;i++)
for(j=0;j<=n-i;j++)
if(i>j) answer+=f[n][i][j],answer%=mod;
}
cout<< (ALL-answer-1+mod)%mod <<endl;
}
signed main(){
solve();
}
浙公网安备 33010602011771号