leetcode102,107. 두 갈래 나무의 층이 두루 다니다

2919 단어 leetcode
두 갈래 트리를 정해서 차원별로 훑어보는 노드 값을 되돌려줍니다.(즉, 왼쪽에서 오른쪽으로 모든 노드에 층층이 접근한다.)
예를 들어, 지정된 두 갈래 트리:[3,9,20,null,null,15,7] ,
    3
   / \
  9  20
    /  \
   15   7

다음 단계를 반복한 결과를 반환합니다.
[
  [3],
  [9,20],
  [15,7]
]

 

생각:


대기열을 빌려 씁니다. 이 제목은 두 개의vector로 처리해야 하며, 줄마다vector는 다음 층을 설치해야 하기 때문에, 층차를 하나하나 옮겨 놓는while 순환을 해야 합니다.
#include 
#include 
#include 

using namespace std;
struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

vector> levelOrder(TreeNode* root) {
	vector> myvec;
	queue q;
	TreeNode * tmp=root;
	if (tmp==NULL)
		return myvec;

	int level=0;
	q.push(tmp);
	while(!q.empty())
	{
		vector tmpvec;
		int size=q.size();
		while(size--)// while, vec 
		{
			tmp=q.front();
			tmpvec.push_back(tmp->val);
			q.pop();
			if (tmp->left)
				q.push(tmp->left);
			if(tmp->right)
				q.push(tmp->right);		
		}
		myvec.push_back(tmpvec);
	}
	return myvec;
}

int main()
{
	TreeNode *root=&TreeNode(1);
	root->left=&TreeNode(2);
	root->right=&TreeNode(3);
	root->left->left=&TreeNode(4);
	root->left->right=&TreeNode(5);
	root->right->left=&TreeNode(6);
	root->right->right=&TreeNode(7);
	root->left->left->left=&TreeNode(8);
	root->left->right->right=&TreeNode(9);
	root->right->left->right=&TreeNode(10);
	vector> myvec;
	myvec=levelOrder(root);
	for (int j=0;j

 
 
107번:
반대로reverse 함수를 사용하면 됩니다. 헤더 파일algorithm
/**
 * 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> levelOrderBottom(TreeNode* root) {
        vector> myvec;
        queue q;
        TreeNode * tmp=root;
        if (tmp==NULL)
            return myvec;

        int level=0;
        q.push(tmp);
        while(!q.empty())
        {
            vector tmpvec;
            int size=q.size();
            while(size--)
            {
                tmp=q.front();
                tmpvec.push_back(tmp->val);
                q.pop();
                if (tmp->left)
                    q.push(tmp->left);
                if(tmp->right)
                    q.push(tmp->right);		
            }
            myvec.push_back(tmpvec);
        }
        reverse(myvec.begin(),myvec.end());// 
        return myvec;
    }
};

좋은 웹페이지 즐겨찾기