我们如何使用 fenwick 树找到数组中非递减子序列的最大和?例如我们有 1 4 4 2 2 3 3 1,这里非递减子序列的最大和是 11 (1 2 2 3 3)。
问问题
1440 次
1 回答
1
可以使用动态规划算法找到最大和。扫描数组并将每个元素的值添加到有效的最大子序列和(子序列以不大于该元素的值结束)。
有效的实现需要某种方法来快速找到给定子范围内的最大值。可以使用增强二叉搜索树来做到这一点。Fenwick 树只是增强二叉搜索树的一种有效实现。Fenwick 树最常见的用途是在某个子范围内查找值的总和。微不足道的修改允许使用它来找到子范围最大值(这是因为在这种特殊情况下,Fenwick 树中的值永远不会减少)。
有关详细信息,请参阅此 Python 代码:
array = [1, 4, 4, 2, 2, 3, 3, 1]
numbers = sorted(set(array))
n = len(numbers)
indexes = {numbers[v]:v+1 for v in range(0, n)}
n += 1
bit = [0] * n
result = 0
for x in array:
pos = indexes[x]
i = pos
maximum = 0
while i != 0:
maximum = max(maximum, bit[i])
i = i & (i-1)
x += maximum
i = pos
while i < n:
bit[i] = max(bit[i], x)
i += i & -i
result = max(result, x)
print(result)
indexes
字典用于将 Fenwick 树的大小从输入数组中的最大数减小到数组的大小。首先嵌套while
在 Fenwick 树中查找子范围最大值。在更新总和之一后,第二个嵌套while
更新 Fenwick 树。
此代码仅适用于正数数组。在一般情况下,输入数组应通过过滤掉所有非正数进行预处理。
时间复杂度为 O(N log N)。空间复杂度为 O(N)。
于 2013-03-04T09:22:26.740 回答