【树】树的广度遍历(1.打成一行 2.按层打成普通多行 3.按层打成之字形多行)

目录:

1.【树】二叉树的非递归遍历(栈&Morris)

2.【树】二叉树转为双向链表

3.【树】树的广度遍历(1.打成一行 2.按层打成普通多行 3.按层打成之字形多行)




1.从上往下打印二叉树:(一行)

打印结果:

a  b  c  d  e  f  g  h  i

vector<int> PrintFromTopToBottom(TreeNode *root) {
        vector<int> ans;
        if(root == NULL)
            return ans;
        queue<TreeNode*> nodes;
        nodes.push(root);
        
        while(!nodes.empty()) {
            TreeNode *curr = nodes.front();
            nodes.pop();
            ans.push_back(curr->val);
            if(curr->left)
                nodes.push(curr->left);
            if(curr->right)
                nodes.push(curr->right);
        }
        return ans;
    }


2.把二叉树打印成普通多行

打印结果:

a

b  c

d  e  f  g

h  i


思路:需要知道啥时候换行(所以1.两个变量 记录 当前层&下一层的节点数;2.队列 因为全都是从左向右打印

vector<vector<int> > Print(TreeNode* pRoot) {
	vector<vector<int> > result;
    if(pRoot == NULL)
        return result;
            
    queue<TreeNode*> nodes;
	nodes.push(pRoot);
	vector<int> tmp;//记录每一层的答案,换层时需清空
	int nextLineNum = 0, currLineNum = 1;
            
	while(!nodes.empty()) {
                
		TreeNode* currNode = nodes.front();
		nodes.pop();
		tmp.push_back(currNode->val);
		currLineNum--;
                
		if(currNode->left) {
			nodes.push(currNode->left);
			nextLineNum++;
		}
		if(currNode->right) {
			nodes.push(currNode->right);
				nextLineNum++;
		}
                
		if(currLineNum == 0) {
			currLineNum = nextLineNum;
			nextLineNum = 0;
			result.push_back(tmp);
			while(!tmp.empty())
				tmp.pop_back();
		}
    }//while结束
    return result;
}

3.按之字形顺序打印二叉树

打印结果:

a

c  b

d  e  f  g

i  h

思路:

不仅需要知道啥时候换行,还需要知道 从左向右 还是 从右向左。所以:

1.用两个栈,根据层数判断:先压入右孩子还是左孩子

2.无需再像上题用两个变量记录节点数,直接用 两个栈都不empty()即可

vector<vector<int> > Print(TreeNode* pRoot) {
	vector<vector<int> > result;
	if(pRoot == NULL)
		return result;
        
	stack<TreeNode*> nodes[2];
	int level = 0;
	nodes[level].push(pRoot);
	vector<int> tmp;
        
	while(!nodes[0].empty() || !nodes[1].empty()) {
        TreeNode* currNode = nodes[level].top();
        nodes[level].pop();
        tmp.push_back(currNode->val);
         
        if(level == 0) {//先压入左孩子
            if(currNode->left) 
             	nodes[1 - level].push(currNode->left);

           	if(currNode->right) 
               	nodes[1 - level].push(currNode->right);
            }
        else {//先压入右孩子
           	if(currNode->right) 
               	nodes[1 - level].push(currNode->right);

            if(currNode->left) 
               	nodes[1 - level].push(currNode->left);
        }
            
        if(nodes[level].empty()) {//结束当前层
            result.push_back(tmp);
            while(!tmp.empty()) {
                tmp.pop_back();
            }

            level = 1 - level;
        }
    }//while结束
    return result; 
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值