두 갈래 나무 관련 문제

5370 단어
제목: 두 갈래 트리에서 한 노드에서 다음 노드를 찾습니다
1. 부모 포인터가 있습니다.두 가지 상황으로 나누다
Node* FindNextNode(Node* node)
    {
        if (_root == NULL || node == NULL)
            return NULL;
        Node* next = NULL;
        //1、 , 
        if (node->_right != NULL)
        {
            Node* right = node->_right;
            while (right->_left)
                right = right->_left;
            next = right;
        }
        //2、 , , , 
        else
        {
            Node* parent = node->_parent;
            Node* cur = node;
            // parent parent , 
            while (parent && cur == parent->_right)
            {
                cur = parent;
                parent = parent->_parent;
            }
            next = parent;
        }
        return next;
    }

2. 일반 두 갈래 나무, 중서를 빌려 두루 다니다
// 
    Node* FindNextNode(Node* node)
    {
        if (_root == NULL || node == NULL)
            return NULL;
        bool find = false;
        stack s;
        Node* cur = _root;
        while (cur || !s.empty())
        {
            while (cur)
            {   
                s.push(cur);
                cur = cur->_LeftChild;
            }
            Node* top = s.top();
            s.pop();
            if (find)
                return top;
            if (top == node)
                find = true;
            cur = top->_RightChild;
        }
        if (s.empty())
            cout << " " << endl;
        return NULL;
    }

제목: 두 갈래 나무의 균형 여부를 판단한다(왼쪽 나무와 오른쪽 나무의 높이는 1을 넘지 않는다)
    bool Isbalance()
    {
        int depth = 0;
        return _Isbalance(_root,depth);
    }

// , 
// , 。 
    bool _Isbalance(Node*& root, int& depth)
    {
        if (root == NULL)
        {
            depth = 0;
            return true;
        }
        int left = 0;
        int right = 0;
        if (_Isbalance(root->_left, left) && _Isbalance(root->_right, right))
        {
            if (abs(left - right) <= 1)
            {
                depth = (left > right ? left + 1 : right + 1);
                return true;
            }
        }
        return false;
    }

좋은 웹페이지 즐겨찾기