leetcode_637_두 갈래 나무의 층 평균치

1696 단어
비공 두 갈래 나무를 정하고 각 층 노드의 평균값으로 구성된 그룹을 되돌려줍니다.
예 1:
 :
    3
   / \
  9  20
    /  \
   15   7
 : [3, 14.5, 11]
 :
 0  3,   1  14.5,  2  11.   [3, 14.5, 11].

참고:
  • 노드 값의 범위는 32비트 기호 정수 범위 내에..
  • /**
     * DEFINITION FOR A BINARY TREE NODE.
     * STRUCT TREENODE {
     *     INT VAL;
     *     TREENODE *LEFT;
     *     TREENODE *RIGHT;
     *     TREENODE(INT X) : VAL(X), LEFT(NULL), RIGHT(NULL) {}
     * };
     */
    class Solution {
    public:
        vector averageOfLevels(TreeNode* root) {
            vector ret;
            queue que;
            que.push(root);// 
            while(!que.empty()) // , 1, 0;
            {
                int size = que.size(); // 
                double level = 0;
                for(int i = 0; i < size; ++i) 
                {
                    TreeNode* node = que.front(); // 
                    que.pop(); // 
                    level += node->val;
                    if(node->left) que.push(node->left); 
                    if(node->right) que.push(node->right);
                }
                ret.push_back(level/size);
            }
            return ret;
            
        }
    };

    문제의 예시에서 우리는 제목이 두 갈래 나무의 각 층의 노드를 훑어보고 그 평균치를 계산해야 한다는 것을 대충 알 수 있다.두 갈래 트리의 검색은 이 몇 가지 방법과 다름없다. 먼저 훑어보기, 중간 훑어보기, 뒤로 훑어보기, BFS, DFS.제목과 결합하면 BFS가 제목에 비교적 적합한 방법임이 분명하다.
    우리는 먼저 대기열에push 루트 노드를 이동한 다음, 대기열이 비어 있지 않을 때, 전체 대기열을 훑어봅니다.대기열의 노드를 훑어볼 때마다 값을 꺼내서 팝업합니다.이 때 왼쪽 하위 노드가 비어 있는지 확인하고, 비어 있지 않으면 하위 노드push를 대기열에 넣습니다.이렇게 해서 마침 한 층의 노드를 두루 훑어보았는데, 다음 층의 노드는 모두 대열에 저장되었다.한 층씩 마지막 층까지 순환하면 문제는 순조롭게 해결된다.

    좋은 웹페이지 즐겨찾기