使用 C++ 檢查二叉樹是否按層級排序


在這裡,我們將瞭解如何檢查二叉樹是否按層級排序。按層級排序的二叉樹如下所示:

每一層中的節點從左到右排序,並且每一層都包含比其上一層更高的值。

我們可以透過執行層序遍歷來解決此問題,並跟蹤當前層的最小值和最大值。使用另一個變數 prev_max 來儲存上一層的最大值。然後比較當前層的最小值和上一層的最大值 prev_max。如果最小值大於 prev_max,則樹在當前層之前按層級排序。然後更新 prev_max 並更新當前層最大值。重複此操作,直到遍歷完所有層。

示例

#include <iostream>
#include <queue>
using namespace std;
class Node {
   public:
   int key;
   Node *left, *right;
};
Node* getNode(int key) {
   Node* newNode = new Node;
   newNode->key = key;
   newNode->left = newNode->right = NULL;
   return newNode;
}
bool isLevelWiseSorted(Node* root) {
   int prevMax = INT_MIN;
   int min_val, max_val;
   int levelSize;
   queue<Node*> q;
   q.push(root);
   while (!q.empty()) {
      levelSize = q.size();
      min_val = INT_MAX;
      max_val = INT_MIN;
      while (levelSize > 0) {
         root = q.front();
         q.pop();
         levelSize--;
         min_val = min(min_val, root->key);
         max_val = max(max_val, root->key);
         if (root->left)
         q.push(root->left);
         if (root->right)
         q.push(root->right);
      }
      if (min_val <= prevMax)
         return false;
      prevMax = max_val;
   }
   return true;
}
int main() {
   Node* root = getNode(1);
   root->left = getNode(2);
   root->right = getNode(3);
   root->left->left = getNode(4);
   root->left->right = getNode(5);
   root->right->left = getNode(6);
   root->right->right = getNode(7);
   if (isLevelWiseSorted(root))
      cout << "Tree is levelwise Sorted";
   else
      cout << "Tree is Not levelwise sorted";
}

輸出

Tree is level wise Sorted

更新於: 2019-10-22

89 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.