题解 Math teacher's homework

题目传送门

题目大意

给出 \(n,k\) 以及 \(a_{1,2,...,n}\) ，求有多少个 \(m_{1,2,...,n}\) 满足 \(\forall i,m_i\le a_i\) 且 \(\oplus_{i=1}^{n} m_i=k\) 。

\(n\le 50,a_i\le 2^{31}-1\)

思路

这个题目真的很神仙。。。

首先你要想到一点，就是对于二进制下的数，肯定是前面一段都相同，突然某一位 \(a_i=1\) 你 \(m_i=0\) 那么 \(m_i\) 你后面就可以乱选了。然后根据这个我们可以设状态 \(dp[i][len][pre]\) 表示到第 \(i\) 个数，你前面 \(len\) 位不能乱选，其余可以乱选，异或前缀和为 \(pre\) 的方案数。具体转移见代码，自认为理解定义之后就可以理解转移了。然后你发现空间开不下，但是实际上你发现你确定 \(len\) 之后 \(pre\) 前面 \(len-1\) 位就确定了，所以状态可以优化到 \(2\) 。具体见代码。

\(\texttt{Code}\)

#include <bits/stdc++.h>
using namespace std;

#define Int register int
#define mod 1000000003

template <typename T> inline void read (T &t){t = 0;char c = getchar();int f = 1;while (c < '0' || c > '9'){if (c == '-') f = -f;c = getchar();}while (c >= '0' && c <= '9'){t = (t << 3) + (t << 1) + c - '0';c = getchar();} t *= f;}
template <typename T,typename ... Args> inline void read (T &t,Args&... args){read (t);read (args...);}
template <typename T> inline void write (T x){if (x < 0){x = -x;putchar ('-');}if (x > 9) write (x / 10);putchar (x % 10 + '0');}

int mul (int a,int b){return 1ll * a * b % mod;}
int dec (int a,int b){return a >= b ? a - b : a + mod - b;}
int add (int a,int b){return a + b >= mod ? a + b - mod : a + b;}

int n,m,a[55],pw[35],dp[55][35][2];

int dfs (int i,int pre,int len){
	pre &= (~((1 << len) - 1));
	if (i > n) return !pre;
	int k = (pre & (1 << len)) ? 1 : 0,res = 0,now = 0;
	if (~dp[i][len][k]) return dp[i][len][k];
	for (Int j = 31;~j;-- j)
		if (a[i] & (1 << j)){
			res = add (res,mul (pw[min (len,j)],dfs (i + 1,pre ^ now,max (len,j))));
			now |= (1 << j);
		}
	return dp[i][len][k] = res;
}

signed main(){
	pw[0] = 1;
	for (Int i = 1;i <= 31;++ i) pw[i] = (pw[i - 1] << 1) % mod;
	while (~scanf ("%d%d",&n,&m) && (n || m)){
		memset (dp,-1,sizeof (dp));
		for (Int i = 1;i <= n;++ i) read (a[i]),a[i] ++;
		write (dfs (1,m,0)),putchar ('\n');
	}
 	return 0;
}