144. Binary Tree Preorder Traversal

這個(gè)系列有preorder inorder postorder三題,考點(diǎn)都不是遞歸,而是迭代。迭代挺難想的,尤其是外面的while循環(huán)。這題的迭代寫(xiě)法我模仿inorder寫(xiě)出來(lái)了,一步步在紙上模擬入棧出棧就好了。

RECURSION:

    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        return dfs(root, res);
    }

    private List<Integer> dfs(TreeNode root, List<Integer> res) {
        if (root == null) {
            return res;
        }
        res.add(root.val);
        dfs(root.left, res);
        dfs(root.right, res);
        return res;
    }

ITERATION:

public List<Integer> preorderTraversal(TreeNode root) {
    ArrayList<Integer> res = new ArrayList<>();
    Stack<TreeNode> s = new Stack<>();
    while (!s.isEmpty() || root!=null) {
        if (root != null) {
            res.add(root.val);
            s.push(root);
            root = root.left;
        }else {
            TreeNode node = s.pop();
            root = node.right;
        }
    }
    return res;
}

看了下discussion,很多跟我寫(xiě)的都不一樣,說(shuō)明迭代寫(xiě)法確實(shí)比遞歸寫(xiě)法多,按照自己的思路來(lái)就好。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容