二叉树预订

时间:2020-11-12 14:30:49

标签: c++ data-structures return binary-tree preorder

我的预订方法从主要方法不起作用,但是数组元素已经正确形成(我在打印每一步时都进行了检查)。定义中使用的所有功能/方法均能正常工作。我认为问题是由于“ return elem;”​​行引起的。任何人都可以提出任何想法,这太重要了。

T* TreeNode<T>::preorder()const 
{
    T* elem = new T[elCount(left)+elCount(right)+1];
    elem[0] = data;
    TreeNode<T>* curr = left;
    Stack<TreeNode<T>*> st;
    if (right) {
        st.push(right);
    }
    int i = 1;
    while (curr) {
        elem[i++] = curr->getData();
        //std::cout << elem[i - 1];
        
        if (curr->getRight()) 
            st.push(curr->getRight());
        
        curr = curr->getLeft();
        if (!curr) 
        {
            curr = st.getTop();
            st.pop();
        }   }
    return elem;
}

1 个答案:

答案 0 :(得分:1)

很难准确地看到您未指定的内容,但是似乎您正在尝试从二进制树中获取所有数据。 This GFG article对如何横穿二叉树进行了非常简单的解释和细分。显然,您的节点更复杂,但是总体逻辑是相同的。我想问一问关于preorder()函数是否确实需要成为树节点类的一部分,以及一个单独的函数是否会更有用。我还考虑使用std::vector而不是您当前使用的手纸。如果您需要固定大小,则std::array也可能会更好。

要显示如何重写代码,可以尝试类似的

std::vector<int> vect;

template<class T>
void preorder(const TreeNode<T>& node, vector<T>& elem) {
    if(!node.getData()) {
        return;
    }

    elem.push(node.getData());

    if(node.getLeft()) {
        preorder(node.getLeft());
    }

    if(node.getRight()) {
        preorder(node.getRight());
    }
}

这不是完美的,因为我是在脑海中写下来的,但这应该为遍历二叉树和提取所有数据提供简单的基础。一点递归应该使它容易得多。

相关问题