C ++中n元树中的下一个更大的元素

在本教程中,我们将编写一个程序来查找 n 叉树中的下一个较大元素

让我们看看解决问题的步骤。

  • 检查当前节点数据是否大于预期元素。

  • 如果是,则检查结果是否为空或结果是否大于当前节点数据。

  • 如果满足上述条件,则更新结果。

  • 获取当前节点子节点。

  • 迭代孩子。

    • 调用递归函数。

示例

让我们看看代码。

#include <bits/stdc++.h>
using namespace std;
struct Node {
   int data;
   vector<Node*> child;
};
Node* newNode(int data) {
   Node* newNode = new Node;
   newNode->data = data;
   return newNode;
}
void findNextGreaterElement(Node* root, int x, Node** result) {
   if (root == NULL) {
      return;
   }
   if (root->data > x) {
      if (!(*result) || (*result)->data > root->data) {
         *result = root;
      }
   }
   int childCount = root->child.size();
   for (int i = 0; i < childCount; i++) {
      findNextGreaterElement(root->child[i], x, result);
   }
   return;
}
int main() {
   Node* root = newNode(10);
   root->child.push_back(newNode(12));
   root->child.push_back(newNode(23));
   root->child.push_back(newNode(45));
   root->child[0]->child.push_back(newNode(40));
   root->child[1]->child.push_back(newNode(33));
   root->child[2]->child.push_back(newNode(12));
   Node* result = NULL;
   findNextGreaterElement(root, 20, &result);
   cout << result->data << endl;
   return 0;
}
输出结果

如果你运行上面的代码,那么你会得到下面的结果。

23

结论

如果您对本教程有任何疑问,请在评论部分提及。