码迷,mamicode.com
首页 > 其他好文 > 详细

1102. Invert a Binary Tree (25)

时间:2015-12-06 13:06:49      阅读:224      评论:0      收藏:0      [点我收藏+]

标签:

反转二叉树就是左右节点互换

层遍历使用队列实现

中序遍历递归实现


The following is from Max Howell @twitter:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can‘t invert a binary tree on a whiteboard so fuck off.

Now it‘s your turn to prove that YOU CAN invert a binary tree!

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer N (<=10) which is the total number of nodes in the tree -- and hence the nodes are numbered from 0 to N-1. Then N lines follow, each corresponds to a node from 0 to N-1, and gives the indices of the left and right children of the node. If the child does not exist, a "-" will be put at the position. Any pair of children are separated by a space.

Output Specification:

For each test case, print in the first line the level-order, and then in the second line the in-order traversal sequences of the inverted tree. There must be exactly one space between any adjacent numbers, and no extra space at the end of the line.

Sample Input:
8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6
Sample Output:
3 7 2 6 4 0 5 1
6 5 7 4 3 2 0 1

 

  1. #include <iostream>
  2. #include <queue>
  3. using namespace std;
  4. int n;
  5. int lfchild[10] = { -1 };
  6. int richild[10] = { -1 };
  7. queue<int> tree;
  8. int cnt = 0;
  9. void Inorder(int a) {
  10. if (lfchild[a] != -1)
  11. Inorder(lfchild[a]);
  12. cout << a;
  13. cnt++;
  14. if (cnt < n)
  15. cout << " ";
  16. if (richild[a] != -1)
  17. Inorder(richild[a]);
  18. }
  19. int main(void) {
  20. for (int i = 0; i < 10; i++) {
  21. lfchild[i] = -1;
  22. richild[i] = -1;
  23. }
  24. int mark[10] = { 0 };
  25. cin >> n;
  26. for (int i = 0; i < n; i++) {
  27. char l, r;
  28. cin >> l >> r;
  29. if (l != ‘-‘) {
  30. richild[i] = l - ‘0‘;
  31. mark[l - ‘0‘]++;
  32. }
  33. if (r != ‘-‘) {
  34. lfchild[i] = r - ‘0‘;
  35. mark[r - ‘0‘]++;
  36. }
  37. }
  38. int root = 0;
  39. for (int i = 0; i < n; i++) {
  40. if (mark[i] == 0) {
  41. root = i;
  42. break;
  43. }
  44. }
  45. tree.push(root);
  46. while (tree.size()>0)
  47. {
  48. if (lfchild[tree.front()] != -1) {
  49. tree.push(lfchild[tree.front()]);
  50. }
  51. if (richild[tree.front()] != -1) {
  52. tree.push(richild[tree.front()]);
  53. }
  54. cout << tree.front();
  55. tree.pop();
  56. if (tree.size() != 0)
  57. cout << " ";
  58. }
  59. cout << endl;
  60. Inorder(root);
  61. return 0;
  62. }





1102. Invert a Binary Tree (25)

标签:

原文地址:http://www.cnblogs.com/zzandliz/p/5023353.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!