C++ 中兩個 BST 節點之間的最大元素


問題陳述

給定一個 N 元素陣列和兩個整數 A、B,它們屬於給定陣列。透過從 arr[0] 到 arr[n-1] 插入元素來建立一個二叉搜尋樹。任務是從 A 到 B 的路徑中找到最大元素。

示例

如果陣列為 {24, 23, 15, 36, 19, 41, 25, 35},我們可以形成如下所示的 BST −

如果我們考慮 A = 19 且 B = 41,則這兩個節點之間的最大元素為 41

演算法

  • 找到節點 A 和 B 的最低公共祖先 (LCA)。
  • 在 LCA 和 A 之間找到最大節點。我們稱之為 max1
  • 在 LCA 和 B 之間找到最大節點。我們稱之為 max2
  • 返回 max1 和 max2 的最大值

示例

現在讓我們看一個示例 −

 即時演示

#include <bits/stdc++.h>
using namespace std;
struct node {
   int data;
   struct node* left;
   struct node* right;
};
node *createNode(int x) {
   node *p = new node();
   p -> data = x;
   p -> left = NULL;
   p -> right = NULL;
   return p;
}
void insertNode(struct node *root, int x) {
   node *p = root, *q = NULL;
   while (p != NULL) {
      q = p;
      if (p -> data < x) {
         p = p -> right;
      } else {
         p = p -> left;
      }
   }
   if (q == NULL) {
      p = createNode(x);
   } else {
      if (q -> data < x) {
         q -> right = createNode(x); } else {
            q -> left = createNode(x);
      }
   }
}
int maxelpath(node *q, int x) {
   node *p = q;
   int mx = INT_MIN;
   while (p -> data != x) {
      if (p -> data > x) {
         mx = max(mx, p -> data);
         p = p -> left;
      } else {
         mx = max(mx, p -> data);
         p = p -> right;
      }
   }
   return max(mx, x);
}
int getMaximumElement(struct node *root, int x, int y) {
   node *p = root;
   while ((x < p -> data && y < p -> data) || (x > p ->
   data && y > p -> data)) {
      if (x < p -> data && y < p -> data) {
         p = p -> left;
      } else if (x > p -> data && y > p -> data) {
         p = p -> right;
      }
   }
   return max(maxelpath(p, x), maxelpath(p, y));
}
int main() {
   int arr[] = {24, 23, 15, 36, 19, 41, 25, 35}; int a = 19, b = 41;
   int n = sizeof(arr) / sizeof(arr[0]);
   struct node *root = createNode(arr[0]);
   for (int i = 1; i < n; i++) insertNode(root, arr[i]);
   cout << "Maximum element = " << getMaximumElement(root, a, b) << endl;
   return 0;
}

輸出

Maximum element = 41

更新日期: 2019 年 12 月 31 日

198 次瀏覽

啟動您的職業

完成課程,獲得認證

開始
廣告
© . All rights reserved.