GlAz_C的博客

原创 二叉树两节点的最短距离(一次dfs完成)

字节面试问到的，leetcode上没看到，记录一下自己的做法此题近似于leetcode 236.二叉树的最近公共祖先，但有较大不同做法为：对二叉树进行dfs遍历，递归函数写法为func dfs(root,p,q *TreeNode,ans *int) int，其中root为当前遍历的节点，p,q为要求的两个节点，ans是答案的指针，返回int值总体思想为利用返回值表征两个节点中的一个是否出现过，并记录距当层的高度，若根节点、左子树、右子树中的二者同时出现p,q则更新答案分多种情况讨论：