牛客 和与或 数位dp

和与或

题目链接

大致题意:

给出序列R,求满足0<=A[i]<=R[i] 且A[0]+A[1]+…+A[N-1]=A[0] or A[1]… or A[N-1] 的序列个数,答案对1e9+9取模

解题思路:

定义sum=A[0]+A[1]+…+A[N-1],转化为二进制后满足A[0] or A[1]… or A[N-1],表示将A[0] A[1]… A[N-1]转化为二进制后,sum二进制上的1只能来自其中的一个A[i],且该位上不能有超过两个1出现,以此为基础考虑数位dp

状态表示: f[i][j]表示枚举到sum二进制的第i位,原序列的限制情况为j的方案数

第一维都理解,那第二维的限制是什么?

举例来说sum=1110(二进制), R[i]=1010(二进制),如果sum的第四位1是A[i]贡献的,说明A[i]>=1000,那么对于sum的第三位1,A[i]就受到了限制,因为A[i]<=R[i],如果第三位的1 A[i]贡献了,那么A[i]必然会大于R[i],不满足条件. 但是如果sum的第四位1 A[i]不贡献,那么对于sum的第三位1第二位1,A[i]就可以不受限制,可以贡献可以不贡献. 这就是f数组第二维限制的意思

转移方程:枚举到第pos位,限制状态是limit,初始时都是受限制的

limit时每一个R[i]的二进制状态表示,1表示不受限制,0表示受限制,举例来说 limit=1001,如果sum 第pos位选1,那么R[1]和R[4]可以贡献1

​ 如果sum第pos位取0,那么在pos位是1的R[i]就可以在之后的搜索中不受限制

​ 如果sum第pos位取1,有两种情况:

​ 1.R[i]没有被限制,那么R[i]保持不受限制,其他的R在pos位也有1的也不受限制

​ 2.R[i]被限制,但pos位有1,那么R[i]保持继续受限制,其他的R在pos位有1的不受限制

​ (之前不受限制,在后面也不会受限制)

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int mod = 1e9 + 9;
int n;
ll r[13];
int f[63][1 << 12];
ll dfs(int pos, int limit) {
	if (pos < 0)return 1;
	if (f[pos][limit] != -1)return f[pos][limit];
	int state = 0;								//pos位可以选哪些r[i]
	for (int i = 1; i <= n; ++i)
		if (r[i] & (1ll << pos))
			state |= (1 << i);
	ll res = 0;
	res += dfs(pos - 1, state | limit);			//pos位取0
	for (int i = 1; i <= n; ++i) {				//枚举这个1填在哪个数上面
		if ((1ll << i) & limit)					//r[i]不受限制
			res += dfs(pos - 1, state | limit);
		else if (1ll << i & state)				//r[i]受限制,但除了r[i]外有不受限制的
			res += dfs(pos - 1, (state | limit) ^ (1ll << i));
	}
	return f[pos][limit] = res % mod;
}
int main(void)
{
	memset(f, -1, sizeof f);
	cin >> n;
	for (int i = 1; i <= n; ++i)cin >> r[i];
	dfs(62, 0);
	cout << f[62][0] << endl;
	return 0;
}