樹非遞歸遍歷

樹的非遞歸代碼(Java實現(xiàn))

建樹:

import java.util.ArrayDeque;
import java.util.Queue;
import java.util.Scanner;

public class TreeBuilder {
    /**
     * 從文件讀取數(shù)據(jù)按層次遍歷的方式創(chuàng)建一個二叉樹 (測試用)
     * 文件輸入只能是'#'或數(shù)字,'#'表示null
     * 測試用例1: 1 2 3 4 5 # # 6 # 7 9
     * 測試用例2: 1 2 3 4 5
     *
     * @return
     */
    public TreeNode<Integer> createTree(Scanner in) {
        String val = in.next();
        if (val.equals("#")) {
            return null;
        }
        TreeNode<Integer> node = new TreeNode<>(Integer.valueOf(val));
        TreeNode<Integer> root = node;
        Queue<TreeNode<Integer>> queue = new ArrayDeque<>();
        queue.add(node);
        while (!queue.isEmpty()) {
            int len = queue.size();
            for (int i = 0; i < len; i++) {
                node = queue.remove();
                if (!in.hasNext()) {
                    break;
                }
                val = in.next();
                if (val.equals("#")) {
                    node.left = null;
                } else {
                    node.left = new TreeNode<>(Integer.valueOf(val));
                    queue.add(node.left);
                }
                if (!in.hasNext()) {
                    break;
                }
                val = in.next();
                if (val.equals("#")) {
                    node.right = null;
                } else {
                    node.right = new TreeNode<>(Integer.valueOf(val));
                    queue.add(node.right);
                }
            }
        }
        return root;
    }
}

樹節(jié)點定義:

public class TreeNode<T> {
    TreeNode(T val) {
        this.val = val;
    }
    T val;
    TreeNode<T> left;
    TreeNode<T> right;
}

層次遍歷都许、前中后遍歷的代碼:

import java.util.*;

public class TreeIter<T> {

    /**
     * 層次遍歷,用隊列實現(xiàn)
     *
     * @param root
     * @return
     */
    public List<T> levelOrder(TreeNode<T> root) {
        List<T> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Queue<TreeNode<T>> queue = new ArrayDeque<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode<T> node = queue.remove();
                res.add(node.val);
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        return res;
    }

    /**
     * 前序遍歷非遞歸
     *
     * @param root
     * @return
     */
    public List<T> preOrder(TreeNode<T> root) {
        List<T> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        TreeNode<T> curNode = root;
        Stack<TreeNode<T>> stack = new Stack<>();
        while (curNode != null || !stack.isEmpty()) {
            while (curNode != null) {
                stack.add(curNode);
                res.add(curNode.val);
                curNode = curNode.left;
            }
            curNode = stack.pop();
            curNode = curNode.right;
        }
        return res;
    }

    /**
     * 中序遍歷非遞歸
     *
     * @param root
     * @return
     */
    public List<T> midOrder(TreeNode<T> root) {
        List<T> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        Stack<TreeNode<T>> stack = new Stack<>();
        TreeNode<T> curNode = root;
        while (curNode != null || !stack.isEmpty()) {
            while (curNode != null) {
                stack.push(curNode);
                curNode = curNode.left;
            }
            curNode = stack.pop();
            res.add(curNode.val);
            curNode = curNode.right;
        }
        return res;
    }

    /**
     * 后序遍歷非遞歸(雙棧法,好理解)
     *
     * @param root
     * @return
     */
    public List<T> postOrder(TreeNode<T> root) {
        List<T> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Stack<TreeNode<T>> stackRes = new Stack<>();
        Stack<TreeNode<T>> stackTmp = new Stack<>();
        stackTmp.push(root);
        TreeNode<T> curNode;

        while (!stackTmp.isEmpty()) {
            curNode = stackTmp.pop();
            stackRes.push(curNode);
            if (curNode.left != null) {
                stackTmp.add(curNode.left);
            }
            if (curNode.right != null) {
                stackTmp.add(curNode.right);
            }
        }

        while (!stackRes.isEmpty()) {
            res.add(stackRes.pop().val);
        }
        return res;
    }

    /**
     * 后序遍歷非遞歸
     * <p>
     * 參考鏈接,注釋非常詳細
     * http://www.cnblogs.com/ybwang/archive/2011/10/04/lastOrderTraverse.html
     *
     * @param root
     * @return
     */
    public List<T> postOrder2(TreeNode<T> root) {
        List<T> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        Stack<TreeNode<T>> stack = new Stack<>();
        stack.push(root);
        TreeNode<T> curNode = null;
        TreeNode<T> preNode = null;
        while (!stack.isEmpty()) {
            curNode = stack.peek();
            if (preNode == null || preNode.left == curNode || preNode.right == curNode) {
                if (curNode.left != null) {
                    stack.push(curNode.left);
                } else if (curNode.right != null) {
                    stack.push(curNode.right);
                } else {
                    res.add(stack.pop().val);
                }
            } else if (curNode.left == preNode) {
                if (curNode.right != null) {
                    stack.push(curNode.right);
                } else {
                    res.add(stack.pop().val);
                }
            } else if (curNode.right == preNode) {
                res.add(stack.pop().val);
            }
            preNode = curNode;
        }
        return res;
    }
}

測試用例:
input.txt

1 2 3 4 5 # # 6 # 7 9

測試代碼:

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.List;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) throws FileNotFoundException {
        TreeIter<Integer> treeIter = new TreeIter<>();
        Scanner in = new Scanner(new FileInputStream("input.txt"));
        TreeNode<Integer> root = new TreeBuilder().createTree(in);

        List<Integer> preOrder = treeIter.preOrder(root);
        System.out.println(preOrder);

        List<Integer> midOrder = treeIter.midOrder(root);
        System.out.println(midOrder);

        List<Integer> postOrder = treeIter.postOrder(root);
        System.out.println(postOrder);

        List<Integer> postOrder2 = treeIter.postOrder2(root);
        System.out.println(postOrder2);

        List<Integer> levelOrder = treeIter.levelOrder(root);
        System.out.println(levelOrder);
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市白粉,隨后出現(xiàn)的幾起案子传泊,更是在濱河造成了極大的恐慌,老刑警劉巖鸭巴,帶你破解...
    沈念sama閱讀 216,470評論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件或渤,死亡現(xiàn)場離奇詭異,居然都是意外死亡奕扣,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,393評論 3 392
  • 文/潘曉璐 我一進店門掌敬,熙熙樓的掌柜王于貴愁眉苦臉地迎上來惯豆,“玉大人,你說我怎么就攤上這事奔害】蓿” “怎么了?”我有些...
    開封第一講書人閱讀 162,577評論 0 353
  • 文/不壞的土叔 我叫張陵华临,是天一觀的道長芯杀。 經(jīng)常有香客問我,道長雅潭,這世上最難降的妖魔是什么揭厚? 我笑而不...
    開封第一講書人閱讀 58,176評論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮扶供,結(jié)果婚禮上筛圆,老公的妹妹穿的比我還像新娘。我一直安慰自己椿浓,他們只是感情好太援,可當我...
    茶點故事閱讀 67,189評論 6 388
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著扳碍,像睡著了一般提岔。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上笋敞,一...
    開封第一講書人閱讀 51,155評論 1 299
  • 那天碱蒙,我揣著相機與錄音,去河邊找鬼夯巷。 笑死振亮,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的鞭莽。 我是一名探鬼主播坊秸,決...
    沈念sama閱讀 40,041評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼澎怒!你這毒婦竟也來了褒搔?” 一聲冷哼從身側(cè)響起阶牍,我...
    開封第一講書人閱讀 38,903評論 0 274
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎星瘾,沒想到半個月后走孽,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,319評論 1 310
  • 正文 獨居荒郊野嶺守林人離奇死亡琳状,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,539評論 2 332
  • 正文 我和宋清朗相戀三年磕瓷,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片念逞。...
    茶點故事閱讀 39,703評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡困食,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出翎承,到底是詐尸還是另有隱情硕盹,我是刑警寧澤,帶...
    沈念sama閱讀 35,417評論 5 343
  • 正文 年R本政府宣布叨咖,位于F島的核電站瘩例,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏甸各。R本人自食惡果不足惜垛贤,卻給世界環(huán)境...
    茶點故事閱讀 41,013評論 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望趣倾。 院中可真熱鬧南吮,春花似錦、人聲如沸誊酌。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,664評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽碧浊。三九已至涂邀,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間箱锐,已是汗流浹背比勉。 一陣腳步聲響...
    開封第一講書人閱讀 32,818評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留驹止,地道東北人浩聋。 一個月前我還...
    沈念sama閱讀 47,711評論 2 368
  • 正文 我出身青樓,卻偏偏與公主長得像臊恋,于是被迫代替她去往敵國和親衣洁。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,601評論 2 353

推薦閱讀更多精彩內(nèi)容