大意:
有M个开关,控制着N个灯,触碰一下开关,会把他所控制的灯的状态改变。对于M个开关你可以选择碰一次或者不碰,求最后开着多少个灯的三次方的期望值E(X^3)*2^m。
题解:这题我觉得思想挺好的。首先是问题转化。我们知道E(x^3)是所有(一共2^m)状态最后留下的灯的数量的三次方,加和除以2^m。所以乘2^m之后应该就是求灯数三次方的和。X^3 = (x1 + x2 + x3 + x4 + … + xn)^3,其中xi表示当前状态p下有没有被打开。E(X^3) * 2^m其实就是把X分解成3块,单独考虑xi,xj,xk,(i,j,k∈[1,2,…n]),如果在2^m中的某个p状态, xi、xj、xk同为1那么对答案贡献1,再跑一个压位dp就好了。
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
using namespace std;
const int MAX_N = 102;
const int MOD = 1e9 + 7;
long long sm[MAX_N];
long long f[MAX_N][8];
long long rev_bit(long long mask, int i, int j, int k, long long p) {
if ((1LL << i) & p) mask ^= 1;
if ((1LL << j) & p) mask ^= 2;
if ((1LL << k) & p) mask ^= 4;
return mask;
}
void sub_main() {
int n, m;
scanf("%d%d", &n, &m);
memset(sm, 0, sizeof(sm));
for (int i = 1; i <= m; ++i) {
int k = 0;
scanf("%d", &k);
sm[i] = 0;
for (int j = 0; j < k; ++j) {
int x;
scanf("%d", &x);
sm[i] = sm[i] | (1LL << (x-1));
}
}
long long ans = 0;
for (int i = 0; i < n; ++i)
for (int j = 0; j < n; ++j)
for (int k = 0; k < n; ++k) {
memset(f, 0, sizeof(f));
f[0][0] = 1;
for (int l = 1; l <= m; ++l)
for (int mask = 0; mask < 8; ++mask) {
(f[l][rev_bit(mask, i, j, k, sm[l])] += f[l-1][mask]) %= MOD;
(f[l][mask] += f[l-1][mask]) %= MOD;
}
(ans += f[m][7]) %= MOD;
}
cout << ans << endl;
}
int main() {
int t;
scanf("%d", &t);
for (int i = 1; i <= t; ++i) {
printf("Case #%d: ", i);
sub_main();
}
}