Giả sử chúng ta có một danh sách các số được gọi là num, chúng ta phải kiểm tra xem nó có đại diện cho một maxheap hay không. Chúng tôi sẽ tuân theo các quy tắc này -
- nums [i] =nums [2 * i + 1] khi 2 * i + 1 nằm trong dải ô
- nums [i] =nums [2 * i + 2] khi 2 * i + 2 nằm trong dải ô
Vì vậy, nếu đầu vào là [5, 3, 4, 1, 2], thì đầu ra sẽ là True
Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -
- đối với tôi trong phạm vi từ 0 đến (kích thước của nums) / 2, hãy thực hiện
- nếu nums [i]> =nums [2 * i + 1] không đúng, thì
- trả về Sai
- nếu i * 2 + 2 <=(kích thước của nums) -1, thì
- nếu nums [i]> =nums [2 * i + 2] không đúng, thì
- trả về Sai
- nếu nums [i]> =nums [2 * i + 2] không đúng, thì
- nếu nums [i]> =nums [2 * i + 1] không đúng, thì
- trả về True
Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -
Ví dụ
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))
Đầu vào
[5, 3, 4, 1, 2]
Đầu ra
True