java棧實現二叉樹的非遞歸遍歷的示例代碼

Posted on by

一般來說遍歷二叉樹用到遞歸，但是用Stack進行遍歷也是一個不錯的方法。

二叉樹設置

class Node{
	public int val;
	public Node left;
	public Node right;
	
	public Node(int v) {
		val=v;
		left=null;
		right=null;
	}
}

public class Main {
	public static void main(String[] args) {
		Node head =new Node(0);
		Node node1=new Node(1);
		Node node2=new Node(2);
		Node node3=new Node(3);
		Node node4=new Node(4);
		Node node5=new Node(5);
		Node node6=new Node(6);
		head.left=node1;
		head.right=node2;
		node1.left=node3;
		node1.right=node4;
		node2.left=node5;
		node2.right=node6;
		pos2(head);
		pos1(head);
		in(head);
	}

在這裡插入圖片描述

前序遍歷

思想和流程

  • 彈出就打印
  • 如果有右子樹，就壓入
  • 如果有左子樹，就壓入
public static void pos1(Node h) {
	System.out.print("先序遍歷 ");
	if(h!=null) {
		Stack<Node>stack =new Stack<Node>();
		stack.push(h);
		while(!stack.isEmpty()) {
			h=stack.pop();
			System.out.print(h.val+" ");
			if(h.right!=null) {
				stack.push(h.right);
			}
			if(h.left!=null) {
				stack.push(h.left);
			}
		}
	}
	System.out.println();
}

後序遍歷

前序遍歷的順序是父節點，左，右，而後序遍歷的順序是左，右，父節點，也就是說前序遍歷左右替換之後就是後序遍歷的倒過來。所以隻要把前序遍歷時候左右子節點壓棧的順序調換一下，再用另一個棧儲存，然後再彈出就是後序遍歷瞭。在此代碼就不多寫瞭。

中序遍歷

思路和流程

  • 彈出就打印
  • 整條左邊界依次入棧
  • 上一條件無法繼續，彈出打印，開始右子樹
public static void in(Node h) {
	System.out.print("中序遍歷 ");
	if(h!=null) {
		Stack<Node>stack =new Stack<Node>();
		while(!stack.isEmpty()||h!=null) {
			if(h!=null) {
				stack.push(h);
				h=h.left;
			}else {
				h=stack.pop();
				System.out.print(h.val+" ");
				h=h.right;
			}
		}
	}
	System.out.println();
}

後序遍歷變體

用一個Stack實現
思路是左節點沒處理就處理左節點，左節點處理過瞭就處理右節點，右節點處理完瞭就返回。

public static void pos2(Node h) {
	if(h!=null) {
		Stack<Node>stack =new Stack<Node>();
		stack.push(h);
		Node c=null;//用c記錄上一次被打印的節點
		while(!stack.isEmpty()) {
			c=stack.peek();
			if(c.left!=null&&h!=c.left&&h!=c.right) {
				stack.push(c.left);
			}else if(c.right!=null&&h!=c.right) {
				stack.push(c.right);
			}else {
				System.out.print(stack.pop().val+" ");
				h=c;//記錄本次被打印的節點
			}
		}
	}
}

打印結果

在這裡插入圖片描述

到此這篇關於java棧實現二叉樹的非遞歸遍歷的文章就介紹到這瞭,更多相關java實現二叉樹的非遞歸遍歷內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：