Python 中有最大堆吗?

pythonserver side programmingprogramming更新于 2025/12/25 18:52:17

假设我们有一个名为 nums 的数字列表，我们必须检查它是否代表最大堆。我们将遵循这些规则 −

当 2*i + 1 在范围内时，nums[i] = nums[2*i + 1]
当 2*i + 2 在范围内时，nums[i] = nums[2*i + 2]

因此，如果输入为 [5, 3, 4, 1, 2]，则输出将为 True

为了解决这个问题，我们将遵循以下步骤 −

对于范围为 0 到 (nums 的大小)/2 的 i，执行
- 如果 nums[i] >= nums[2*i+1] 不为真，则
  - 返回 False
- 如果i*2+2 <= (nums 的大小)-1，则
  - 如果 nums[i] >= nums[2*i+2] 不成立，则
    - 返回 False
返回 True

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

示例

class Solution:
   def solve(self, nums):
      for i in range(len(nums)//2):
         if not nums[i] >= nums[2*i+1]:
            return False
         if i*2+2 <= len(nums)-1:
            if not nums[i] >= nums[2*i+2]:
               return False
      return True
ob = Solution()
nums = [5, 3, 4, 1, 2]
print(ob.solve(nums))