从上到下打印二叉树(C++)

//从上往下打印出二叉树的每个节点,同层节点从左至右打印。
//文件夹中有visio解释图

#include
#include
#include
using namespace std;

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

class Solution {
public:
    vector PrintFromTopToBottom(TreeNode* root) {
        queue que;
		vector vec;
		que.push(root);//先将整个二叉树放入队列
		while(!que.empty()) //当队列非空进行循环
		{
			TreeNode* p;
			p = que.front();//先读取队列的首元素
			que.pop();//弹出队列的首元素
			if(p == NULL)
				continue;//所有元素存入vec后,由于队列中存放着空指针,依然进入循环,但此时p的值为NULL,不执行下面的操作,跳出循环结束
			que.push(p->left);
			que.push(p->right);
			vec.push_back(p->val);
		}
		return vec;

    }
};

void CreateTree(TreeNode **T)
{
	//TreeNode *T;
	//T = S;
	char c;
	cin>>c;

	if('#' == c)
		*T = NULL;//当遇到#时,令树的根节点为NULL,从而结束该分支的递归
	else
	{
		(*T) = (TreeNode*)malloc(sizeof(TreeNode));
		(*T)->val = c;//给根节点赋值
		CreateTree(&(*T)->left);//创建左子树
		CreateTree(&(*T)->right);//创建右子树
	}
}

int main()
{
    TreeNode *T;
	T = (TreeNode*)malloc(sizeof(TreeNode));
	
	CreateTree(&T);
	Solution so;
	so.PrintFromTopToBottom(T);
    
    return 0;
}

 

你可能感兴趣的:(C++)