Problem Arrangement ZOJ - 3777(状压dp + 期望)

时间:2022-11-11 21:29:52

ZOJ - 3777

就是一个入门状压dp期望

dp[i][j]

当前状态为i,分数为j时的情况数
然后看代码 有注释
#include <iostream>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <map>
#include <cctype>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <list>
#include <cmath>
#include <bitset>
#define rap(i, a, n) for(int i=a; i<=n; i++)
#define rep(i, a, n) for(int i=a; i<n; i++)
#define lap(i, a, n) for(int i=n; i>=a; i--)
#define lep(i, a, n) for(int i=n; i>a; i--)
#define rd(a) scanf("%d", &a)
#define rlld(a) scanf("%lld", &a)
#define rc(a) scanf("%c", &a)
#define rs(a) scanf("%s", a)
#define rb(a) scanf("%lf", &a)
#define rf(a) scanf("%f", &a)
#define pd(a) printf("%d\n", a)
#define plld(a) printf("%lld\n", a)
#define pc(a) printf("%c\n", a)
#define ps(a) printf("%s\n", a)
#define MOD 2018
#define LL long long
#define ULL unsigned long long
#define Pair pair<int, int>
#define mem(a, b) memset(a, b, sizeof(a))
#define _ ios_base::sync_with_stdio(0),cin.tie(0)
//freopen("1.txt", "r", stdin);
using namespace std;
const int maxn = , INF = 0x7fffffff;
int n, m;
int str[][];
int vis[];
int f[];
int dp[ << ][]; //当前状态为i,分数为j时的情况数 int gcd(int a, int b)
{
return b == ? a : gcd(b, a % b);
} int main()
{
f[] = ;
for(int i = ; i <= ; i++)
f[i] = f[i - ] * i; int T;
rd(T);
while(T--)
{
mem(dp, );
rd(n), rd(m);
for(int i = ; i < n; i++)
{
for(int j = ; j < n; j++)
rd(str[i][j]);
} int cnt = ;
dp[][] = ;
for(int i = ; i < ( << n); i++) //遍历所有的状态
{
cnt = ;
for(int j = ; j < n; j++) //计算当前已经选择了前多少个
if(i & ( << j)) cnt++;
for(int j = ; j < n; j++) //当前第cnt + 1个应该放在哪一个位置
{
if(i & ( << j)) continue;
for(int k = ; k <= m; k++) //因为我们不知道前几个的具体位置 所以就不知道具体数值 只能遍历所有(m最大才500)
{
if(k + str[cnt][j] >= m) dp[i | ( << j)][m] += dp[i][k];
else dp[i | ( << j)][k + str[cnt][j]] += dp[i][k];
}
}
}
if(dp[( << n) - ][m] == )
printf("No solution\n");
else
{ int d = gcd(dp[( << n) - ][m], f[n]);
printf("%d/%d\n", f[n]/d, dp[( << n) - ][m]/d); } } return ;
}