【谷歌面試題】有序輸出兩棵二叉查詢樹中的元素

WalkingInTheWind發表於2013-05-28

題目:給出兩棵二叉查詢樹,有序輸出所有元素,時間複雜度O(n),空間複雜度O(h),h為樹的高度

此題就是把兩棵二叉查詢樹的中序遍歷過程結合在一起。

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

void print2BSTsInSortedOrder(TreeNode *root1, TreeNode *root2)
{
	stack<TreeNode *> stk1, stk2;
	TreeNode *p1 = root1;
	while(p1)
	{
		stk1.push(p1);
		p1 = p1->left;
	}
	TreeNode *p2 = root2;
	while(p2 != NULL)
	{
		stk2.push(p2);
		p2 = p2->left;
	}
	while(!stk1.empty() || !stk2.empty())
	{
		if(!stk1.empty())
			p1 = stk1.top();
		if(!stk2.empty())
			p2 = stk2.top();
		if(p1 == NULL || (p2 && p2->val <= p1->val))
		{
			printf("%d ", p2->val);
			stk2.pop();
			p2 = p2->right;
			while(p2 != NULL)
			{
				stk2.push(p2);
				p2 = p2->left;
			}
		}
		else if(p2 == NULL || (p1 && p1->val <= p2->val))
		{
			printf("%d ", p1->val);
			stk1.pop();
			p1 = p1->right;
			while(p1)
			{
				stk1.push(p1);
				p1 = p1->left;
			}
		}
	}
}



相關文章