leetcode257(バイナリツリーのすべてのパス:バイナリツリー検索)

バイナリツリーを指定すると、ルートノードからリーフノードへのすべてのパスを返します。
説明:葉ノードは、子ノードのないノードを指します。

例:
入力:[1,2,3、null、5]
出力:出力:["1-> 2-> 5"、 "1-> 3"]

解決策:DFSの深さ優先の検索方法を採用し、スタックを使用してトラバースすると同時に、パスがスタックに保存されるだけです。リーフノードが検出されると、スタック内の既存の要素が文字列パスに結合されます。

class Solution {
    
    
    public List<String> binaryTreePaths(TreeNode root) {
    
    
        /*
        * 用栈来保存路径,同时进行树的遍历,
        * 同时要注意的是栈中的每一个结点都有两次搜索机会
        * 当一次搜索得到结点的右孩子后,结点不出栈,而是存储在
        * 栈中,防止路径顺序被破环,直到从栈中提取的该结点,再弹出栈
        */
            Map<TreeNode,Integer>map=new HashMap<>();
            List<String>res=new ArrayList<>();
            Stack<TreeNode>stack=new Stack<>();
            TreeNode node=root;
            while (!stack.isEmpty() || node != null) {
    
    
                while (node != null) {
    
    
                    stack.push(node);
                    map.put(node,0);
                    if (node.left == null && node.right == null) {
    
    
                        res.add(getRoute(stack));
                    }
                    node = node.left;
                }
                node = stack.peek();
                int temp=map.get(node);
                if(temp==1) {
    
    
                    stack.pop();
                    node=null;
                }
                else {
    
    
                    map.put(node, 1);
                    node=node.right;
                }
            }
            return res;
    }
    private String getRoute(Stack<TreeNode>stack){
    
    
        StringBuilder route=new StringBuilder();
        for(TreeNode x:stack){
    
    
            route.append(x.val).append("->");
        }
        int len=route.length();
        route.delete(len-2,len);
        return route.toString();
    }
}

おすすめ

転載: blog.csdn.net/CY2333333/article/details/108397118