和与或
大致题意:
给出序列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;
}