leetcode[110] Balanced Binary Tree

5580 단어 LeetCode
나무 한 그루가 균형 잡힌 두 갈래 나무인지 아닌지를 판단하다.
사고방식: 귀착.
각 노드의 좌우 나무는 균형 두 갈래 나무이며, 좌우 나무의 높이 차이는 1을 넘지 않는다.
/**

 * Definition for binary tree

 * struct TreeNode {

 *     int val;

 *     TreeNode *left;

 *     TreeNode *right;

 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}

 * };

 */

class Solution {

public:

    int heightTree(TreeNode *root)

    {

        if (!root) return 0;

        int lf = 0, ri = 0;

        if (root -> left)

            lf = heightTree(root -> left);

        if (root -> right)

            ri = heightTree(root -> right);

        return max(lf, ri) + 1;

    }

    

    bool isBalanced(TreeNode *root) {

        if (!root) return true;

        bool lf, ri;

        lf = isBalanced(root -> left);

        ri = isBalanced(root -> right);

        return lf && ri && (abs(heightTree(root->left)-heightTree(root->right))<=1);

    }

};

사고방식2: 중서를 이용하여 각 노드에 대해 좌자수 우자수 고도의 차이치를 판단한다.
/**

 * Definition for binary tree

 * struct TreeNode {

 *     int val;

 *     TreeNode *left;

 *     TreeNode *right;

 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}

 * };

 */

class Solution {

public:

    int heightTree(TreeNode *root)

    {

        if (!root) return 0;

        int lf = 0, ri = 0;

        if (root -> left)

            lf = heightTree(root -> left);

        if (root -> right)

            ri = heightTree(root -> right);

        return max(lf, ri) + 1;

    }

    

    bool isBalanced(TreeNode *root) {

        if (!root) return true;

        

        stack<TreeNode *> sta;

        TreeNode *p = root;

        

        while(p || !sta.empty())

        {

            while(p)

            {

                sta.push(p);

                p = p -> left;

            }

            if (!sta.empty())

            {

                p = sta.top();

                sta.pop();

                if (abs(heightTree(p -> left) - heightTree(p -> right)) > 1)

                    return false;

                p = p -> right;

            }

        }

        return true;

    }

};

좋은 웹페이지 즐겨찾기