题目
n n n个人,一个人有 n n n个菜,依次拿菜,每个菜的喜爱值为 a i j a_{ij} aij,拿走概率为 a i j ∑ j = 1 n a i j \frac{a_{ij}}{\sum_{j=1}^{n}a_{ij}} ∑j=1naijaij,求所有菜被拿走的概率
思路
状压
a
n
s
[
i
]
[
j
]
ans[i][j]
ans[i][j]表示拿了
i
i
i种菜,拿到其中
j
j
j类型这个菜的概率
f
[
x
]
f[x]
f[x]表示拿的状态为
x
x
x这种的总概率
代码
在这里插入代码片#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f;
int a[1010][1010],ans[2000010][30],f[2001000],inv[10010],sum[1000];
const int mod = 998244353;
int quick_pow(int a,int b){
int res=1;
while(b){
if(b&1) res=res*a%mod;
a=a*a%mod;
b>>=1;
}
return res%mod;
}
signed main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int n;
cin>>n;
inv[1]=1;
for(int i=2;i<=3000;i++){
inv[i]=mod-mod/i*inv[mod%i]%mod;
}
for(int i=1;i<=n;i++)
for(int j=0;j<n;j++) cin>>a[i][j],sum[i]+=a[i][j];
f[0]=1;
for(int i=1;i<(1<<n);i++){
int cnt=0;
for(int j=0;j<n;j++){
if(i&(1<<j)) cnt++;
}
int tot=0;
for(int j=0;j<n;j++){
if(i&(1<<j)){
tot=(tot+a[cnt][j])%mod;
}
}
for(int j=0;j<n;j++){
if(i&&(1<<j)){
ans[cnt][j]=(ans[cnt][j]+f[i^(1<<j)]*a[cnt][j]%mod*inv[sum[cnt]-tot+a[cnt][j]]%mod)%mod;
f[i]=(f[i]+f[i^(1<<j)]*a[cnt][j]%mod*inv[sum[cnt]-tot+a[cnt][j]]%mod)%mod;
}
}
}
for(int i=1;i<=n;i++){
for(int j=0;j<n;j++){
cout<<ans[i][j];
if(j==n-1){
if(i!=n) cout<<endl;
}
else cout<<" ";
}
}
return 0;
}