[编程之美]分层遍历二叉树

编程之美提供了两个方法：递归和用vector来保存遍历结果；但是第二种用vector的方法需要保存下所有节点，是o(n)的空间复杂度。

下面是自己的一个的方法，利用与BFS类似的思想，借助一个栈可以使空间复杂度降到 O(M)，M表示具有最多节点的那层的节点数，最坏情况下 M = N/2；

 

void BFSTree(node* tree)
{
	if (! tree) return;
	queue<node*> Q;
	Q.push(tree);
	int num=1;
	while(!Q.empty())
	{
		int next=0;
		while(num--)
		{
			node* t=Q.front();
			Q.pop();
			cout << t->val<<" ";
			if (t->lchild)
			{
				next++;
				Q.push(t->lchild);
			}
			if (t->rchild)
			{
				next++;
				Q.push(t->rchild);
			}
		}
		cout << endl;
		num=next;
	}
}

int main()
{
	vector<node> tree;
	tree.reverse(9);
	for(int i=1;i<=n;i++)
		tree.push_back(node(i));
	tree[1].lchild=tree[2];
	tree[1].rchild=tree[3];
	tree[2].lchild=tree[4];
	tree[2].rchild=tree[5];
	tree[5].lchild=tree[7];
	tree[5].rchild=tree[8];
	tree[3].rchild=tree[6];

	node* root= tree[1];
	BFSTree(root);
	system("pause");
}