题解 Math teacher's homework

时间:2023-03-10 07:11:21
题解 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;
}