有序输出两棵二叉查找树中的元素
题目:给出两棵二叉查找树,有序输出所有元素,时间复杂度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)
10. {
- 11. stack<TreeNode *> stk1, stk2;
- 12. TreeNode *p1 = root1;
- 13. while(p1)
- 14. {
- 15. stk1.push(p1);
- 16. p1 = p1->left; //最小节点
- 17. }
- 18. TreeNode *p2 = root2;
- 19. while(p2 != NULL)
- 20. {
- 21. stk2.push(p2);
- 22. p2 = p2->left; //最小节点
- 23. }
- 24. while(!stk1.empty() || !stk2.empty())
- 25. {
- 26. if(!stk1.empty())
- 27. p1 = stk1.top();
- 28. if(!stk2.empty())
- 29. p2 = stk2.top();
- 30. if(p1 == NULL || (p2 && p2->val <= p1->val))
- 31. {
- 32. printf("%d ", p2->val);
- 33. stk2.pop();
- 34. //获取p2节点的下一个节点,右节点的最左节点,加入栈中,下次从栈中弹出的就是下一个节点
- 35. p2 = p2->right;
- 36. while(p2 != NULL)
- 37. {
- 38. stk2.push(p2);
- 39. p2 = p2->left;
- 40. }
- 41. }
- 42. else if(p2 == NULL || (p1 && p1->val <= p2->val))
- 43. {
- 44. printf("%d ", p1->val);
- 45. stk1.pop();
- //获取p1节点的下一个节点,右节点的最左节点,加入栈中,下次从栈中弹出的就是下一个节点
- 46. p1 = p1->right;
- 47. while(p1)
- 48. {
- 49. stk1.push(p1);
- 50. p1 = p1->left;
- 51. }
- 52. }
- 53. }
54. }