设N(x, y) 为该问题的解数。显然N(0, 0) 是 1,因为唯一的解是 (0, 0, 0, 0)。如果 x 或 y 为负数,则没有解决方案,因为我们要求 a1、a2、a3、a4 都是非负数。
否则,我们可以继续求解最低位,并生成递归关系。让我们写 n:0 和 n:1 来表示 2n+0 和 2n+1(所以 0 和 1 是最低位)。
然后:
N(0, 0) = 1
N(-x, y) = N(x, -y) = 0
N(x:0, y:0) = N(x, y) + N(x-1, y) + N(x, y-1) + N(x-1, y-1)
N(x:0, y:1) = N(x:1, y:0) = 0
N(x:1, y:1) = 4 * N(x, y)
要查看这些,必须考虑任何 a1、a2、a3、a4 的可能低位。
首先N(x:0, y:0)。我们需要 a1+a2 的低位为 0,这意味着 a1 和 a2 要么都是偶数,要么都是奇数。如果它们都是奇数,则存在进位,并且较高位的总和加上 1 必须与 x 的较高位相加。相同的逻辑适用于 a3、a4。有 4 种可能:a1、a2、a3、a4 的所有低位为 0,a1、a2 的低位为 1,a3、a4 的低位为 1,a1、a2、a3、a4 的低位为 1。 4 例。
其次是N(x:0, y:1) 和N(x:1, y:0)。如果一个和为偶数,另一个为奇数,则没有解决方案:可以检查每个组合的 a1、a2、a3、a4 的最低位以找出答案。
第三个N(x:1, y:1)。 a1 和 a2 中的一个必须是奇数,同样,a3 和 a4 中的一个也必须是奇数。这有 4 种可能性,在任何情况下都没有进位。
这是一个完整的解决方案:
def N(x, y):
if x == y == 0: return 1
if x < 0 or y < 0: return 0
if x % 2 == y % 2 == 0:
return N(x//2, y//2) + N(x//2-1, y//2) + N(x//2, y//2-1) + N(x//2-1, y//2-1)
elif x % 2 == y % 2 == 1:
return 4 * N(x//2, y//2)
else:
return 0
该算法进行多次递归调用,因此理论上是指数级的。但在实践中,许多分支会很快终止,因此代码运行速度足够快,可以处理高达 2^30 的值。当然,你可以添加缓存或者使用动态规划表来保证 O(log(x)+log(y)) 的运行时间。
最后,为了提高正确性的信心,这里有一些针对简单 O(xy) 解决方案的测试:
def N_slow(x, y):
s = 0
for a1 in xrange(x + 1):
for a3 in xrange(y + 1):
a2 = x - a1
a4 = y - a3
if a1 ^ a2 ^ a3 ^ a4:
continue
s += 1
return s
for x in xrange(50):
for y in xrange(50):
n = N(x, y)
ns = N_slow(x, y)
if n != ns:
print 'N(%d, %d) = %d, want %d' % (x, y, n, ns)