450:删除二叉搜索树中的节点
This commit is contained in:
轩辕龙儿
parent
4a52c17ee4
commit
cc87f70069
116
src/main/java/leetcode/editor/cn/DeleteNodeInABst.java
Normal file
116
src/main/java/leetcode/editor/cn/DeleteNodeInABst.java
Normal file
@ -0,0 +1,116 @@
|
||||
//<p>给定一个二叉搜索树的根节点 <strong>root </strong>和一个值 <strong>key</strong>，删除二叉搜索树中的 <strong>key </strong>对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。</p>
|
||||
//
|
||||
//<p>一般来说，删除节点可分为两个步骤：</p>
|
||||
//
|
||||
//<ol>
|
||||
// <li>首先找到需要删除的节点；</li>
|
||||
// <li>如果找到了，删除它。</li>
|
||||
//</ol>
|
||||
//
|
||||
//<p> </p>
|
||||
//
|
||||
//<p><strong>示例 1:</strong></p>
|
||||
//
|
||||
//<p><img src="https://assets.leetcode.com/uploads/2020/09/04/del_node_1.jpg" style="width: 800px;" /></p>
|
||||
//
|
||||
//<pre>
|
||||
//<strong>输入：</strong>root = [5,3,6,2,4,null,7], key = 3
|
||||
//<strong>输出：</strong>[5,4,6,2,null,null,7]
|
||||
//<strong>解释：</strong>给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
|
||||
//一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
|
||||
//另一个正确答案是 [5,2,6,null,4,null,7]。
|
||||
//
|
||||
//<img src="https://assets.leetcode.com/uploads/2020/09/04/del_node_supp.jpg" style="width: 350px;" />
|
||||
//</pre>
|
||||
//
|
||||
//<p><strong>示例 2:</strong></p>
|
||||
//
|
||||
//<pre>
|
||||
//<strong>输入:</strong> root = [5,3,6,2,4,null,7], key = 0
|
||||
//<strong>输出:</strong> [5,3,6,2,4,null,7]
|
||||
//<strong>解释:</strong> 二叉树不包含值为 0 的节点
|
||||
//</pre>
|
||||
//
|
||||
//<p><strong>示例 3:</strong></p>
|
||||
//
|
||||
//<pre>
|
||||
//<strong>输入:</strong> root = [], key = 0
|
||||
//<strong>输出:</strong> []</pre>
|
||||
//
|
||||
//<p> </p>
|
||||
//
|
||||
//<p><strong>提示:</strong></p>
|
||||
//
|
||||
//<ul>
|
||||
// <li>节点数的范围 <code>[0, 10<sup>4</sup>]</code>.</li>
|
||||
// <li><code>-10<sup>5</sup> <= Node.val <= 10<sup>5</sup></code></li>
|
||||
// <li>节点值唯一</li>
|
||||
// <li><code>root</code> 是合法的二叉搜索树</li>
|
||||
// <li><code>-10<sup>5</sup> <= key <= 10<sup>5</sup></code></li>
|
||||
//</ul>
|
||||
//
|
||||
//<p> </p>
|
||||
//
|
||||
//<p><strong>进阶：</strong> 要求算法时间复杂度为 O(h)，h 为树的高度。</p>
|
||||
//<div><div>Related Topics</div><div><li>树</li><li>二叉搜索树</li><li>二叉树</li></div></div><br><div><li>👍 782</li><li>👎 0</li></div>
|
||||
package leetcode.editor.cn;
|
||||
|
||||
import com.code.leet.entiy.TreeNode;
|
||||
|
||||
// 450:删除二叉搜索树中的节点
|
||||
public class DeleteNodeInABst {
|
||||
public static void main(String[] args) {
|
||||
Solution solution = new DeleteNodeInABst().new Solution();
|
||||
// TO TEST
|
||||
}
|
||||
//leetcode submit region begin(Prohibit modification and deletion)
|
||||
|
||||
/**
|
||||
* Definition for a binary tree node.
|
||||
* public class TreeNode {
|
||||
* int val;
|
||||
* TreeNode left;
|
||||
* TreeNode right;
|
||||
* TreeNode() {}
|
||||
* TreeNode(int val) { this.val = val; }
|
||||
* TreeNode(int val, TreeNode left, TreeNode right) {
|
||||
* this.val = val;
|
||||
* this.left = left;
|
||||
* this.right = right;
|
||||
* }
|
||||
* }
|
||||
*/
|
||||
class Solution {
|
||||
public TreeNode deleteNode(TreeNode root, int key) {
|
||||
if (root == null) {
|
||||
return root;
|
||||
}
|
||||
TreeNode node = root;
|
||||
TreeNode left;
|
||||
if (root.val == key) {
|
||||
if (root.right != null) {
|
||||
node = root.right;
|
||||
left = root.left;
|
||||
root = node;
|
||||
while (node.left != null) {
|
||||
node = node.left;
|
||||
}
|
||||
node.left = left;
|
||||
} else {
|
||||
root = root.left;
|
||||
}
|
||||
} else if (root.val > key) {
|
||||
if (root.left != null) {
|
||||
root.left = deleteNode(root.left, key);
|
||||
}
|
||||
} else {
|
||||
if (root.right != null) {
|
||||
root.right = deleteNode(root.right, key);
|
||||
}
|
||||
}
|
||||
return root;
|
||||
}
|
||||
}
|
||||
//leetcode submit region end(Prohibit modification and deletion)
|
||||
|
||||
}
|
||||
@ -0,0 +1,55 @@
|
||||
<p>给定一个二叉搜索树的根节点 <strong>root </strong>和一个值 <strong>key</strong>，删除二叉搜索树中的 <strong>key </strong>对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。</p>
|
||||
|
||||
<p>一般来说，删除节点可分为两个步骤：</p>
|
||||
|
||||
<ol>
|
||||
<li>首先找到需要删除的节点；</li>
|
||||
<li>如果找到了，删除它。</li>
|
||||
</ol>
|
||||
|
||||
<p> </p>
|
||||
|
||||
<p><strong>示例 1:</strong></p>
|
||||
|
||||
<p><img src="https://assets.leetcode.com/uploads/2020/09/04/del_node_1.jpg" style="width: 800px;" /></p>
|
||||
|
||||
<pre>
|
||||
<strong>输入：</strong>root = [5,3,6,2,4,null,7], key = 3
|
||||
<strong>输出：</strong>[5,4,6,2,null,null,7]
|
||||
<strong>解释：</strong>给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
|
||||
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
|
||||
另一个正确答案是 [5,2,6,null,4,null,7]。
|
||||
|
||||
<img src="https://assets.leetcode.com/uploads/2020/09/04/del_node_supp.jpg" style="width: 350px;" />
|
||||
</pre>
|
||||
|
||||
<p><strong>示例 2:</strong></p>
|
||||
|
||||
<pre>
|
||||
<strong>输入:</strong> root = [5,3,6,2,4,null,7], key = 0
|
||||
<strong>输出:</strong> [5,3,6,2,4,null,7]
|
||||
<strong>解释:</strong> 二叉树不包含值为 0 的节点
|
||||
</pre>
|
||||
|
||||
<p><strong>示例 3:</strong></p>
|
||||
|
||||
<pre>
|
||||
<strong>输入:</strong> root = [], key = 0
|
||||
<strong>输出:</strong> []</pre>
|
||||
|
||||
<p> </p>
|
||||
|
||||
<p><strong>提示:</strong></p>
|
||||
|
||||
<ul>
|
||||
<li>节点数的范围 <code>[0, 10<sup>4</sup>]</code>.</li>
|
||||
<li><code>-10<sup>5</sup> <= Node.val <= 10<sup>5</sup></code></li>
|
||||
<li>节点值唯一</li>
|
||||
<li><code>root</code> 是合法的二叉搜索树</li>
|
||||
<li><code>-10<sup>5</sup> <= key <= 10<sup>5</sup></code></li>
|
||||
</ul>
|
||||
|
||||
<p> </p>
|
||||
|
||||
<p><strong>进阶：</strong> 要求算法时间复杂度为 O(h)，h 为树的高度。</p>
|
||||
<div><div>Related Topics</div><div><li>树</li><li>二叉搜索树</li><li>二叉树</li></div></div><br><div><li>👍 782</li><li>👎 0</li></div>
|
||||
Loading…
Reference in New Issue
Block a user