면접 문제 55(二): 균형 두 갈래 나무

3312 단어 검지 Offer
제목 설명은 두 갈래 나무를 입력하여 이 두 갈래 나무가 균형 두 갈래 나무인지 아닌지를 판단한다.방법1: 각 노드의 깊이를 비교한다.
class Solution {
public:
    int getDepth(TreeNode* root)
    {
        if(root == nullptr)
            return 0;
        int leftH = getDepth(root->left);
        int rightH = getDepth(root->right);
        return max(leftH, rightH) + 1;
    }

    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == nullptr)
            return true;
        int left = getDepth(pRoot->left);
        int right = getDepth(pRoot->right);
        int dif = left - right;
        if(dif > 1 || dif < -1)
            return false;
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

방법2: 후속 스트리밍 방식으로 두 갈래 나무의 모든 노드를 훑어본다. 한 노드를 훑어보기 전에 우리는 그것의 좌우 나무를 훑어보았기 때문에 중복 스트리밍은 존재하지 않는다.
class Solution {
public:
    bool isBalanced(TreeNode* root, int& depth)
    {
        if(root == nullptr)
        {
            //depth = 0;
            return true;
        }
        int left = 0, right = 0;
        if(isBalanced(root->left, left) && isBalanced(root->right, right))
        {
            int dif = left - right;
            if(dif <= 1 && dif >= -1)
            {
                depth = max(left, right) + 1;
                return true;
            }
        }
        return false;
    }

    bool IsBalanced_Solution(TreeNode* pRoot) {
        int depth = 0;
        return isBalanced(pRoot, depth);
    }
};

좋은 웹페이지 즐겨찾기