LeetCode——Binary Tree Paths-白红宇

LeetCode——Binary Tree Paths

阅读量：802 次

发布时间：2023-01-31

本文共 1608 字，大约阅读时间需要 5 分钟。

二叉树遍历问题：生成根到叶路径

问题描述

给定一个二叉树，要求返回所有从根节点到叶节点的路径。每条路径用字符串表示，路径之间用“->”连接节点值。

示例

对于以下二叉树：

所有根到叶的路径为：["1->2->5", "1->3"]。

方法思路

我们可以使用深度优先搜索（DFS）来遍历二叉树。当到达叶节点时，记录当前路径并将其添加到结果列表中。为了实现路径的记录和回溯，我们可以使用一个递归方法，维护一个当前路径的列表。在访问叶节点时，将路径转换为字符串并存储；在递归返回时，清理当前路径。

解决代码

import java.util.ArrayList;
import java.util.List;
public class Solution {
    List
   
     paths = new ArrayList<>();
    List
    
      path = new ArrayList<>();
    public List
     
       binaryTreePaths(TreeNode root) {
        paths.clear();
        path.clear();
        getAllPath(root);
        return paths;
    }
    private void getAllPath(TreeNode node) {
        if (node == null) {
            return;
        }
        path.add(node.val);
        if (node.left == null && node.right == null) {
            // 当前路径是完整的，生成字符串
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < path.size(); i++) {
                if (i != 0) {
                    sb.append("->");
                }
                sb.append(path.get(i));
            }
            paths.add(sb.toString());
        }
        // 继续遍历左子树
        if (node.left != null) {
            getAllPath(node.left);
        }
        // 回溯，移除最后一个节点值
        path.remove(path.size() - 1);
    }
}

代码解释

初始化：在binaryTreePaths方法中，清空paths和path列表，准备开始遍历。

递归遍历：getAllPath方法递归地遍历二叉树。首先处理当前节点，如果当前节点为null，则返回。

记录路径：将当前节点的值添加到path列表中。

检查叶节点：如果当前节点没有左子树和右子树，则表示到达了叶节点。此时，使用StringBuilder将路径转换为字符串，并添加到paths列表中。

继续遍历：递归地处理左子树和右子树。

回溯：在递归返回时，移除path列表的最后一个元素，以确保路径信息正确回溯。

这种方法使用了深度优先搜索，确保先遍历完全右子树再回溯，从而正确生成所有根到叶的路径。

转载地址：http://bqgyk.baihongyu.com/

你可能感兴趣的文章