[LeetCode]637. 두 갈래 나무의 층 평균치
3389 단어 데이터 구조와 알고리즘나의 리코드.
제목
, .
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<double> averageOfLevels(TreeNode* root) {
vector<double> results;
queue que;
que.push(root);
int oldNodeCount=1;
int levelCount=1;
int newNodeCount=0;
double levelAverage=0.0;
while(que.size()!=0)
{
TreeNode* node=que.front();
que.pop();
oldNodeCount--;
levelAverage+=node->val;
if(node->left!=NULL)
{
que.push(node->left);
newNodeCount++;
}
if(node->right!=NULL)
{
que.push(node->right);
newNodeCount++;
}
if(oldNodeCount==0)
{
results.push_back(levelAverage/levelCount);
levelCount=newNodeCount;
oldNodeCount=newNodeCount;
newNodeCount=0;
levelAverage=0.0;
}
}
return results;
}
};
생각
이 해법은 차원 반복을 바탕으로 나의 사고방식은 하나의 보조 대기열로 결점을 보존하고 두 변수로 현재 층의 결점 개수와 다음 층의 결점 개수를 보존하는 것이다.대기열 헤드 결점을 반복할 때마다 현재 결점 개수-1, 그리고 이 결점의 하위 노드를 반복합니다. 비어 있지 않으면 대기열을 추가하고 다음 결점 개수+1, 앞층 결점 개수가 0일 때 현재 층이 반복적으로 완성되면 다음 결점 개수는 현재 층 결점 개수로 변하고 다음 층 결점 개수는 0을 계산하여 대기열의 원소 개수가 0이 될 때까지 계속 반복합니다.
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
두 갈래 나무의 깊이가 두루 다니다텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.