Python 中用于检查堆是否形成最大堆的程序

假设我们有一个表示堆树的列表。我们知道堆是完全二叉树。我们需要检查元素是否形成最大堆。我们知道对于最大堆，每个元素都比它的两个子节点大。

因此，如果输入是 nums = [8, 6, 4, 2, 0, 3]，则输出将为True，因为所有元素都比它们的子节点大。

要解决这个问题，我们将遵循以下步骤：

n ：= nums的大小
对于i从0到n-1的范围内，执行以下操作：
- m := i * 2
- num := nums[i]
- 如果m+1
- 如果 num < nums[m+1]，则
  - 返回 False
如果m+2
如果 num < nums[m+2]，则
- 返回 False
- 返回 True

示例

让我们看下面的实现，以便更好地理解。

def solve(nums):
   n = len(nums)
   for i in range(n):
      m = i * 2
      num = nums[i]
      if m + 1 < n:
         if num < nums[m + 1]:
            return False
      if m + 2 < n:
         if num < nums[m + 2]:
            return False
   return True

nums = [8, 6, 4, 2, 0, 3]
print(solve(nums))