LeetCode222 Count Complete Tree Nodes

제목 링크:


https://leetcode.com/problems/count-complete-tree-nodes/

제목 설명:


완전 두 갈래 나무의 노드 개수를 구하다.

제목 분석:


주의해야 할 것은 반복적인 방식으로 노드의 개수를 구하면 시간이 초과되고 시간의 복잡도는 O(n)이다.완전 두 갈래 나무와 만 두 갈래 나무 노드 개수의 특성을 이용하다.만 두 갈래 나무의 노드 개수는 2^h-1이고 h는 나무의 높이이다.먼저 이 완전 두 갈래 나무가 두 갈래 나무로 가득 차 있는지 판단하고, 직접 결과를 되돌려 주는 것이 아니라면, 다시 귀속해서 해답을 구한다.
노드 왼쪽 트리의 가장 왼쪽 깊이가 노드 오른쪽 트리의 가장 오른쪽 깊이와 같으면 이 노드를 루트 노드로 하는 나무는 두 갈래 나무가 가득하고, 그렇지 않으면 이 노드를 루트 노드로 하는 나무의 노드 개수 = 그 왼쪽 트리의 노드 개수 + 그 오른쪽 트리의 노드 개수 + 그 자체이다.

코드:

class Solution {
public:
    int countNodes(TreeNode* root) {
        if(root==NULL){
            return 0;
        }
        int leftHgt=1;
        int rightHgt=1;
        TreeNode* ptr=NULL;
        ptr=root->left;
        while(ptr!=NULL){
            leftHgt++;
            ptr=ptr->left;
        }
        ptr=root->right;
        while(ptr!=NULL){
            rightHgt++;
            ptr=ptr->right;
        }
        if(leftHgt==rightHgt){
            return (1<<leftHgt)-1;
        }
        else{
            return countNodes(root->left)+countNodes(root->right)+1;
        }
    }
};

좋은 웹페이지 즐겨찾기