1. 程式人生 > >CF327E Axis Walking(LG2396 yyy loves Maths VII)

CF327E Axis Walking(LG2396 yyy loves Maths VII)

題目大意:給一個長度為$n(1\leqslant n\leqslant24)$的序列$S$和$k(0\leqslant k\leqslant2)$個數。

求有多少種$S$的排列方式使得其任何一個字首和都不是$k$個數裡的任意一個。

題解:狀壓$DP$,列舉當前選的數的狀態和下一個數,卡常,列舉下一個數的時候不可以直接列舉,要列舉$lowbit$看是從哪個數轉移過來的,洛谷那道題特別卡常,需要開$O(2)$

卡點:

 

C++ Code:

#include <cstdio>
#define maxn 30
#define N (1 << 24 | 5)
const int mod = 1e9 + 7;
int n, k;
int s[maxn], _1, _2;
inline long long fac(int n) {
	long long res = 1;
	for (int i = 2; i <= n; i++) res = res * i % mod;
	return res;
}

int f[N], S[N];
inline void up(int &a, int b) {if ((a += b) >= mod) a -= mod;}
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) scanf("%d", s + i), S[1 << i - 1] = s[i];
	scanf("%d", &k);
	if (!k) return !printf("%lld\n", fac(n));
	switch (k) {
		case 2: scanf("%d", &_2);
		case 1: scanf("%d", &_1);
	}
	int U = 1 << n;
	f[0] = 1;
	for (register int i = 1; i < U; i++) {
		int k = i & -i;
		S[i] = S[i ^ k] + S[k];
		if (S[i] != _1 && S[i] != _2) for (register int j = i; j; j &= j - 1) up(f[i], f[i ^ (j & -j)]);
	}
	printf("%d\n", f[U - 1]);
	return 0;
}