[Java]将二叉树的左右子树交换 非递归实现

[java]  view plain  copy

1. package dataStruct;  
2.   
3. import java.util.Stack;  
4.   
5. /** 
6.  * 将二叉树的左右子树交换 非递归实现 
7.  * @author YangYi 
8.  */  
9. public class SwapTree {  
10.     private static Stack<Node> stack = new Stack<Node>();  
11.   
12.     public static void main(String args[]) {  
13.         Node root = buildTree();  
14.         inOrderVisit(root);  
15.         swapTree(root);  
16.         System.out.println();  
17.         inOrderVisit(root);  
18.     }  
19.   
20.     public static void inOrderVisit(Node root) {  
21.         if (root == null)  
22.             return;  
23.         inOrderVisit(root.left);  
24.         System.out.print(root.data);  
25.         inOrderVisit(root.right);  
26.     }  
27.   
28.     public static void swapTree(Node root) {  
29.         if (root == null)  
30.             return;  
31.   
32.         Node temp = null;  
33.   
34.         stack.push(root);  
35.         while (!stack.isEmpty()) {  
36.             Node node = stack.peek();  
37.   
38.             if (node.left == null && node.right == null) {  
39.                 node.visited = true;  
40.                 stack.pop();  
41.                 continue;  
42.             }  
43.   
44.             if (node.left != null) {  
45.                 if (!node.left.visited) {  
46.                     stack.push(node.left);  
47.                 }  
48.             }  
49.   
50.             if ((node.left == null || node.left.visited) && node.right != null) {  
51.                 if (!node.right.visited) {  
52.                     stack.push(node.right);  
53.                 }  
54.             }  
55.   
56.             if ((node.left == null || node.left.visited)  
57.                     && (node.right == null || node.right.visited)) {  
58.                 temp = node.left;  
59.                 node.left = node.right;  
60.                 node.right = temp;  
61.                 node.visited = true;  
62.                 stack.pop();  
63.             }  
64.         }  
65.     }  
66.   
67.     public static Node buildTree() {  
68.         Node root = new Node();  
69.         root.data = 1;  
70.   
71.         Node temp = new Node();  
72.         temp.data = 2;  
73.         root.left = temp;  
74.   
75.         temp = new Node();  
76.         temp.data = 3;  
77.         root.right = temp;  
78.   
79.         temp = new Node();  
80.         temp.data = 4;  
81.         root.right.right = temp;  
82.   
83.         temp = new Node();  
84.         temp.data = 5;  
85.         root.left.right = temp;  
86.   
87.         temp = new Node();  
88.         temp.data = 6;  
89.         root.left.right.right = temp;  
90.   
91.         return root;  
92.     }  
93. }  
94.   
95. class Node {  
96.     boolean visited = false;  
97.     int data = 0;  
98.     Node left = null;  
99.     Node right = null;  
100. }