863:二叉树中所有距离为 K 的结点(未完成)
This commit is contained in:
parent
6e631a0604
commit
16080bac5b
@ -2,7 +2,9 @@ package com.code.leet.entiy;
|
||||
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.LinkedList;
|
||||
import java.util.List;
|
||||
import java.util.Queue;
|
||||
|
||||
/**
|
||||
* @Author: hyy
|
||||
@ -70,4 +72,43 @@ public class TreeNode {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public TreeNode preTreeNode(List<Integer> list) {
|
||||
if (list == null || list.size() == 0) {
|
||||
return null;
|
||||
}
|
||||
TreeNode root = new TreeNode(list.get(0));
|
||||
Queue<TreeNode> queue = new LinkedList<>();
|
||||
queue.add(root);
|
||||
for (int i = 1; i < list.size(); i = i + 2) {
|
||||
TreeNode node = queue.poll();
|
||||
Integer left = list.get(i);
|
||||
if (node == null) {
|
||||
queue.add(null);
|
||||
queue.add(null);
|
||||
} else {
|
||||
if (left != null) {
|
||||
node.left = new TreeNode(left);
|
||||
}
|
||||
queue.add(node.left);
|
||||
Integer right = i + 1 < list.size() ? list.get(i + 1) : null;
|
||||
if (right != null) {
|
||||
node.right = new TreeNode(right);
|
||||
}
|
||||
queue.add(node.right);
|
||||
}
|
||||
}
|
||||
return root;
|
||||
}
|
||||
|
||||
public TreeNode search(TreeNode root, int target) {
|
||||
if (root == null || root.val == target) {
|
||||
return root;
|
||||
}
|
||||
TreeNode temp = search(root.left, target);
|
||||
if (temp != null) {
|
||||
return temp;
|
||||
}
|
||||
return search(root.right, target);
|
||||
}
|
||||
}
|
||||
|
@ -0,0 +1,116 @@
|
||||
//给定一个二叉树(具有根结点 root), 一个目标结点 target ,和一个整数值 K 。
|
||||
//
|
||||
// 返回到目标结点 target 距离为 K 的所有结点的值的列表。 答案可以以任何顺序返回。
|
||||
//
|
||||
//
|
||||
//
|
||||
//
|
||||
//
|
||||
//
|
||||
// 示例 1:
|
||||
//
|
||||
// 输入:root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2
|
||||
//输出:[7,4,1]
|
||||
//解释:
|
||||
//所求结点为与目标结点(值为 5)距离为 2 的结点,
|
||||
//值分别为 7,4,以及 1
|
||||
//
|
||||
//
|
||||
//
|
||||
//注意,输入的 "root" 和 "target" 实际上是树上的结点。
|
||||
//上面的输入仅仅是对这些对象进行了序列化描述。
|
||||
//
|
||||
//
|
||||
//
|
||||
//
|
||||
// 提示:
|
||||
//
|
||||
//
|
||||
// 给定的树是非空的。
|
||||
// 树上的每个结点都具有唯一的值 0 <= node.val <= 500 。
|
||||
// 目标结点 target 是树上的结点。
|
||||
// 0 <= K <= 1000.
|
||||
//
|
||||
// Related Topics 树 深度优先搜索 广度优先搜索 二叉树
|
||||
// 👍 354 👎 0
|
||||
|
||||
package leetcode.editor.cn;
|
||||
|
||||
import com.code.leet.entiy.TreeNode;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
//863:二叉树中所有距离为 K 的结点
|
||||
public class AllNodesDistanceKInBinaryTree {
|
||||
public static void main(String[] args) {
|
||||
//测试代码
|
||||
Solution solution = new AllNodesDistanceKInBinaryTree().new Solution();
|
||||
TreeNode root = new TreeNode();
|
||||
root = root.preTreeNode(Arrays.asList(3, 5, 1, 6, 2, 0, 8, null, null, 7, 4));
|
||||
|
||||
solution.distanceK(root, root.search(root, 5), 2);
|
||||
}
|
||||
//力扣代码
|
||||
//leetcode submit region begin(Prohibit modification and deletion)
|
||||
|
||||
/**
|
||||
* Definition for a binary tree node.
|
||||
* public class TreeNode {
|
||||
* int val;
|
||||
* TreeNode left;
|
||||
* TreeNode right;
|
||||
* TreeNode(int x) { val = x; }
|
||||
* }
|
||||
*/
|
||||
class Solution {
|
||||
private Stack<TreeNode> stack = new Stack<>();
|
||||
List<Integer> list = new ArrayList<>();
|
||||
|
||||
public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
|
||||
if (k == 0) {
|
||||
return Arrays.asList(target.val);
|
||||
}
|
||||
if (!root.val.equals(target.val)) {
|
||||
serchTarget(root, target.val, true);
|
||||
}
|
||||
int index = 1;
|
||||
while (!stack.isEmpty() && k >= index) {
|
||||
TreeNode node = stack.pop();
|
||||
dfs(node, 0, k - index);
|
||||
index++;
|
||||
}
|
||||
return list;
|
||||
}
|
||||
|
||||
private boolean serchTarget(TreeNode root, int val, boolean bl) {
|
||||
if (root == null) {
|
||||
return true;
|
||||
}
|
||||
if (root.val == val) {
|
||||
return false;
|
||||
}
|
||||
if (bl) {
|
||||
stack.push(root);
|
||||
}
|
||||
bl = serchTarget(root.left, val, bl) || serchTarget(root.right, val, bl);
|
||||
if (bl) {
|
||||
stack.pop();
|
||||
}
|
||||
return bl;
|
||||
}
|
||||
|
||||
private void dfs(TreeNode root, int index, int k) {
|
||||
if (root == null) {
|
||||
return;
|
||||
}
|
||||
if (index == k) {
|
||||
list.add(root.val);
|
||||
return;
|
||||
}
|
||||
dfs(root.left, index + 1, k);
|
||||
dfs(root.right, index + 1, k);
|
||||
}
|
||||
}
|
||||
//leetcode submit region end(Prohibit modification and deletion)
|
||||
|
||||
}
|
@ -0,0 +1,34 @@
|
||||
<p>给定一个二叉树(具有根结点 <code>root</code>), 一个目标结点 <code>target</code> ,和一个整数值 <code>K</code> 。</p>
|
||||
|
||||
<p>返回到目标结点 <code>target</code> 距离为 <code>K</code> 的所有结点的值的列表。 答案可以以任何顺序返回。</p>
|
||||
|
||||
<p> </p>
|
||||
|
||||
<ol>
|
||||
</ol>
|
||||
|
||||
<p><strong>示例 1:</strong></p>
|
||||
|
||||
<pre><strong>输入:</strong>root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2
|
||||
<strong>输出:</strong>[7,4,1]
|
||||
<strong>解释:</strong>
|
||||
所求结点为与目标结点(值为 5)距离为 2 的结点,
|
||||
值分别为 7,4,以及 1
|
||||
|
||||
<img alt="" src="https://s3-lc-upload.s3.amazonaws.com/uploads/2018/06/28/sketch0.png" style="height: 240px; width: 280px;">
|
||||
|
||||
注意,输入的 "root" 和 "target" 实际上是树上的结点。
|
||||
上面的输入仅仅是对这些对象进行了序列化描述。
|
||||
</pre>
|
||||
|
||||
<p> </p>
|
||||
|
||||
<p><strong>提示:</strong></p>
|
||||
|
||||
<ol>
|
||||
<li>给定的树是非空的。</li>
|
||||
<li>树上的每个结点都具有唯一的值 <code>0 <= node.val <= 500</code> 。</li>
|
||||
<li>目标结点 <code>target</code> 是树上的结点。</li>
|
||||
<li><code>0 <= K <= 1000</code>.</li>
|
||||
</ol>
|
||||
<div><div>Related Topics</div><div><li>树</li><li>深度优先搜索</li><li>广度优先搜索</li><li>二叉树</li></div></div>\n<div><li>👍 354</li><li>👎 0</li></div>
|
File diff suppressed because one or more lines are too long
Loading…
Reference in New Issue
Block a user