1. Add build script for Java.

2. Add height limitation for code blocks in extra.css.
3. Fix "节点" to "结点".
This commit is contained in:
krahets 2023-02-07 04:43:52 +08:00
parent b14568151c
commit ecbf2d1560
54 changed files with 457 additions and 1633 deletions

View file

@ -61,7 +61,7 @@ TreeNode *rightRotate(TreeNode *node) {
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -76,7 +76,7 @@ TreeNode *leftRotate(TreeNode *node) {
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -128,7 +128,7 @@ TreeNode *insertHelper(TreeNode *node, int val) {
updateHeight(node); updateHeight(node);
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -186,7 +186,7 @@ TreeNode *removeHelper(TreeNode *node, int val) {
updateHeight(node); updateHeight(node);
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -28,7 +28,7 @@ int *levelOrder(TreeNode *root, int *size) {
while (front < rear) { while (front < rear) {
// 队列出队 // 队列出队
node = queue[front++]; node = queue[front++];
// 保存结点 // 保存结点
arr[index++] = node->val; arr[index++] = node->val;
if (node->left != NULL) { if (node->left != NULL) {
// 左子结点入队 // 左子结点入队

View file

@ -6,11 +6,9 @@
#include "../include/include.hpp" #include "../include/include.hpp"
/** /* 合并左子数组和右子数组 */
* // 左子数组区间 [left, mid]
* [left, mid] // 右子数组区间 [mid + 1, right]
* [mid + 1, right]
*/
void merge(vector<int>& nums, int left, int mid, int right) { void merge(vector<int>& nums, int left, int mid, int right) {
// 初始化辅助数组 // 初始化辅助数组
vector<int> tmp(nums.begin() + left, nums.begin() + right + 1); vector<int> tmp(nums.begin() + left, nums.begin() + right + 1);

View file

@ -9,7 +9,7 @@
/* AVL 树 */ /* AVL 树 */
class AVLTree { class AVLTree {
public: public:
TreeNode* root; // 根 TreeNode* root; // 根
private: private:
/* 更新结点高度 */ /* 更新结点高度 */
void updateHeight(TreeNode* node) { void updateHeight(TreeNode* node) {
@ -27,7 +27,7 @@ private:
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -41,7 +41,7 @@ private:
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -88,7 +88,7 @@ private:
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -133,7 +133,7 @@ private:
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -16,7 +16,7 @@ vector<int> hierOrder(TreeNode* root) {
while (!queue.empty()) { while (!queue.empty()) {
TreeNode* node = queue.front(); TreeNode* node = queue.front();
queue.pop(); // 队列出队 queue.pop(); // 队列出队
vec.push_back(node->val); // 保存结点 vec.push_back(node->val); // 保存结点
if (node->left != nullptr) if (node->left != nullptr)
queue.push(node->left); // 左子结点入队 queue.push(node->left); // 左子结点入队
if (node->right != nullptr) if (node->right != nullptr)

View file

@ -12,7 +12,7 @@ namespace hello_algo.chapter_tree
/* AVL 树 */ /* AVL 树 */
class AVLTree class AVLTree
{ {
public TreeNode? root; // public TreeNode? root; //
/* 获取结点高度 */ /* 获取结点高度 */
public int height(TreeNode? node) public int height(TreeNode? node)
@ -48,7 +48,7 @@ namespace hello_algo.chapter_tree
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -63,7 +63,7 @@ namespace hello_algo.chapter_tree
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -127,7 +127,7 @@ namespace hello_algo.chapter_tree
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -170,7 +170,7 @@ namespace hello_algo.chapter_tree
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -8,7 +8,7 @@ import . "github.com/krahets/hello-algo/pkg"
/* AVL 树 */ /* AVL 树 */
type avlTree struct { type avlTree struct {
// 根 // 根
root *TreeNode root *TreeNode
} }
@ -57,7 +57,7 @@ func rightRotate(node *TreeNode) *TreeNode {
// 更新结点高度 // 更新结点高度
updateHeight(node) updateHeight(node)
updateHeight(child) updateHeight(child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
@ -71,7 +71,7 @@ func leftRotate(node *TreeNode) *TreeNode {
// 更新结点高度 // 更新结点高度
updateHeight(node) updateHeight(node)
updateHeight(child) updateHeight(child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
@ -130,7 +130,7 @@ func insertHelper(node *TreeNode, val int) *TreeNode {
updateHeight(node) updateHeight(node)
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node) node = rotate(node)
// 返回子树的根 // 返回子树的根
return node return node
} }
@ -174,7 +174,7 @@ func removeHelper(node *TreeNode, val int) *TreeNode {
updateHeight(node) updateHeight(node)
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node) node = rotate(node)
// 返回子树的根 // 返回子树的根
return node return node
} }

View file

@ -20,7 +20,7 @@ func levelOrder(root *TreeNode) []int {
for queue.Len() > 0 { for queue.Len() > 0 {
// poll // poll
node := queue.Remove(queue.Front()).(*TreeNode) node := queue.Remove(queue.Front()).(*TreeNode)
// 保存结点 // 保存结点
nums = append(nums, node.Val) nums = append(nums, node.Val)
if node.Left != nil { if node.Left != nil {
// 左子结点入队 // 左子结点入队

View file

@ -74,6 +74,7 @@ public class space_complexity {
/* 平方阶(递归实现) */ /* 平方阶(递归实现) */
static int quadraticRecur(int n) { static int quadraticRecur(int n) {
if (n <= 0) return 0; if (n <= 0) return 0;
// 数组 nums 长度为 n, n-1, ..., 2, 1
int[] nums = new int[n]; int[] nums = new int[n];
System.out.println("递归 n = " + n + " 中的 nums 长度 = " + nums.length); System.out.println("递归 n = " + n + " 中的 nums 长度 = " + nums.length);
return quadraticRecur(n - 1); return quadraticRecur(n - 1);

View file

@ -9,11 +9,9 @@ package chapter_sorting;
import java.util.*; import java.util.*;
public class merge_sort { public class merge_sort {
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
static void merge(int[] nums, int left, int mid, int right) { static void merge(int[] nums, int left, int mid, int right) {
// 初始化辅助数组 // 初始化辅助数组
int[] tmp = Arrays.copyOfRange(nums, left, right + 1); int[] tmp = Arrays.copyOfRange(nums, left, right + 1);

View file

@ -123,7 +123,7 @@ class QuickSortTailCall {
} }
/* 快速排序(尾递归优化) */ /* 快速排序(尾递归优化) */
static void quickSort(int[] nums, int left, int right) { public static void quickSort(int[] nums, int left, int right) {
// 子数组长度为 1 时终止 // 子数组长度为 1 时终止
while (left < right) { while (left < right) {
// 哨兵划分操作 // 哨兵划分操作

View file

@ -10,7 +10,7 @@ import include.*;
/* AVL 树 */ /* AVL 树 */
class AVLTree { class AVLTree {
TreeNode root; // TreeNode root; //
/* 获取结点高度 */ /* 获取结点高度 */
public int height(TreeNode node) { public int height(TreeNode node) {
@ -42,7 +42,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -56,7 +56,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -109,7 +109,7 @@ class AVLTree {
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -146,7 +146,7 @@ class AVLTree {
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -18,7 +18,7 @@ public class binary_tree_bfs {
List<Integer> list = new ArrayList<>(); List<Integer> list = new ArrayList<>();
while (!queue.isEmpty()) { while (!queue.isEmpty()) {
TreeNode node = queue.poll(); // 队列出队 TreeNode node = queue.poll(); // 队列出队
list.add(node.val); // 保存结点 list.add(node.val); // 保存结点
if (node.left != null) if (node.left != null)
queue.offer(node.left); // 左子结点入队 queue.offer(node.left); // 左子结点入队
if (node.right != null) if (node.right != null)

View file

@ -4,11 +4,9 @@
* Author: IsChristina (christinaxia77@foxmail.com) * Author: IsChristina (christinaxia77@foxmail.com)
*/ */
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
function merge(nums, left, mid, right) { function merge(nums, left, mid, right) {
// 初始化辅助数组 // 初始化辅助数组
let tmp = nums.slice(left, right + 1); let tmp = nums.slice(left, right + 1);

View file

@ -11,7 +11,7 @@ const { printTree } = require("../include/PrintUtil");
class AVLTree { class AVLTree {
/*构造函数*/ /*构造函数*/
constructor() { constructor() {
this.root = null; //根 this.root = null; //根
} }
/* 获取结点高度 */ /* 获取结点高度 */
@ -44,7 +44,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -58,7 +58,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -108,7 +108,7 @@ class AVLTree {
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -141,7 +141,7 @@ class AVLTree {
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -15,7 +15,7 @@ function hierOrder(root) {
let list = []; let list = [];
while (queue.length) { while (queue.length) {
let node = queue.shift(); // 队列出队 let node = queue.shift(); // 队列出队
list.push(node.val); // 保存结点 list.push(node.val); // 保存结点
if (node.left) if (node.left)
queue.push(node.left); // 左子结点入队 queue.push(node.left); // 左子结点入队
if (node.right) if (node.right)

View file

@ -43,7 +43,7 @@ class AVLTree:
# 更新结点高度 # 更新结点高度
self.__update_height(node) self.__update_height(node)
self.__update_height(child) self.__update_height(child)
# 返回旋转后子树的根 # 返回旋转后子树的根
return child return child
""" 左旋操作 """ """ 左旋操作 """
@ -56,7 +56,7 @@ class AVLTree:
# 更新结点高度 # 更新结点高度
self.__update_height(node) self.__update_height(node)
self.__update_height(child) self.__update_height(child)
# 返回旋转后子树的根 # 返回旋转后子树的根
return child return child
""" 执行旋转操作,使该子树重新恢复平衡 """ """ 执行旋转操作,使该子树重新恢复平衡 """

View file

@ -13,7 +13,7 @@ from include import *
""" Driver Code """ """ Driver Code """
if __name__ == "__main__": if __name__ == "__main__":
""" 初始化二叉树 """ """ 初始化二叉树 """
# 初始化 # 初始化
n1 = TreeNode(val=1) n1 = TreeNode(val=1)
n2 = TreeNode(val=2) n2 = TreeNode(val=2)
n3 = TreeNode(val=3) n3 = TreeNode(val=3)
@ -29,7 +29,7 @@ if __name__ == "__main__":
""" 插入与删除结点 """ """ 插入与删除结点 """
P = TreeNode(0) P = TreeNode(0)
# 在 n1 -> n2 中间插入点 P # 在 n1 -> n2 中间插入点 P
n1.left = P n1.left = P
P.left = n2 P.left = n2
print("\n插入结点 P 后\n") print("\n插入结点 P 后\n")

View file

@ -18,7 +18,7 @@ def hier_order(root: Optional[TreeNode]):
res = [] res = []
while queue: while queue:
node = queue.popleft() # 队列出队 node = queue.popleft() # 队列出队
res.append(node.val) # 保存点值 res.append(node.val) # 保存点值
if node.left is not None: if node.left is not None:
queue.append(node.left) # 左子结点入队 queue.append(node.left) # 左子结点入队
if node.right is not None: if node.right is not None:

View file

@ -4,11 +4,9 @@
* Author: nuomi1 (nuomi1@qq.com) * Author: nuomi1 (nuomi1@qq.com)
*/ */
/** /* */
* // [left, mid]
* [left, mid] // [mid + 1, right]
* [mid + 1, right]
*/
func merge(nums: inout [Int], left: Int, mid: Int, right: Int) { func merge(nums: inout [Int], left: Int, mid: Int, right: Int) {
// //
let tmp = Array(nums[left ..< (right + 1)]) let tmp = Array(nums[left ..< (right + 1)])

View file

@ -8,7 +8,7 @@ import utils
/* AVL */ /* AVL */
class AVLTree { class AVLTree {
fileprivate var root: TreeNode? // fileprivate var root: TreeNode? //
/* */ /* */
func height(node: TreeNode?) -> Int { func height(node: TreeNode?) -> Int {
@ -40,7 +40,7 @@ class AVLTree {
// //
updateHeight(node: node) updateHeight(node: node)
updateHeight(node: child) updateHeight(node: child)
// //
return child return child
} }
@ -54,7 +54,7 @@ class AVLTree {
// //
updateHeight(node: node) updateHeight(node: node)
updateHeight(node: child) updateHeight(node: child)
// //
return child return child
} }
@ -112,7 +112,7 @@ class AVLTree {
updateHeight(node: node) // updateHeight(node: node) //
/* 2. 使 */ /* 2. 使 */
node = rotate(node: node) node = rotate(node: node)
// //
return node return node
} }
@ -155,7 +155,7 @@ class AVLTree {
updateHeight(node: node) // updateHeight(node: node) //
/* 2. 使 */ /* 2. 使 */
node = rotate(node: node) node = rotate(node: node)
// //
return node return node
} }

View file

@ -14,7 +14,7 @@ func hierOrder(root: TreeNode) -> [Int] {
var list: [Int] = [] var list: [Int] = []
while !queue.isEmpty { while !queue.isEmpty {
let node = queue.removeFirst() // let node = queue.removeFirst() //
list.append(node.val) // list.append(node.val) //
if let left = node.left { if let left = node.left {
queue.append(left) // queue.append(left) //
} }

View file

@ -4,11 +4,9 @@
* Author: Justin (xiefahit@gmail.com) * Author: Justin (xiefahit@gmail.com)
*/ */
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
function merge(nums: number[], left: number, mid: number, right: number): void { function merge(nums: number[], left: number, mid: number, right: number): void {
// 初始化辅助数组 // 初始化辅助数组
let tmp = nums.slice(left, right + 1); let tmp = nums.slice(left, right + 1);

View file

@ -12,7 +12,7 @@ class AVLTree {
root: TreeNode; root: TreeNode;
/*构造函数*/ /*构造函数*/
constructor() { constructor() {
this.root = null; //根 this.root = null; //根
} }
/* 获取结点高度 */ /* 获取结点高度 */
@ -45,7 +45,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -59,7 +59,7 @@ class AVLTree {
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
@ -113,7 +113,7 @@ class AVLTree {
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -151,7 +151,7 @@ class AVLTree {
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }

View file

@ -16,7 +16,7 @@ function hierOrder(root: TreeNode | null): number[] {
const list: number[] = []; const list: number[] = [];
while (queue.length) { while (queue.length) {
let node = queue.shift() as TreeNode; // 队列出队 let node = queue.shift() as TreeNode; // 队列出队
list.push(node.val); // 保存结点 list.push(node.val); // 保存结点
if (node.left) { if (node.left) {
queue.push(node.left); // 左子结点入队 queue.push(node.left); // 左子结点入队
} }

View file

@ -10,7 +10,7 @@ pub fn AVLTree(comptime T: type) type {
return struct { return struct {
const Self = @This(); const Self = @This();
root: ?*inc.TreeNode(T) = null, // root: ?*inc.TreeNode(T) = null, //
mem_arena: ?std.heap.ArenaAllocator = null, mem_arena: ?std.heap.ArenaAllocator = null,
mem_allocator: std.mem.Allocator = undefined, // mem_allocator: std.mem.Allocator = undefined, //
@ -59,7 +59,7 @@ pub fn AVLTree(comptime T: type) type {
// //
self.updateHeight(node); self.updateHeight(node);
self.updateHeight(child); self.updateHeight(child);
// //
return child; return child;
} }
@ -73,7 +73,7 @@ pub fn AVLTree(comptime T: type) type {
// //
self.updateHeight(node); self.updateHeight(node);
self.updateHeight(child); self.updateHeight(child);
// //
return child; return child;
} }
@ -132,7 +132,7 @@ pub fn AVLTree(comptime T: type) type {
self.updateHeight(node); // self.updateHeight(node); //
// 2. 使 // 2. 使
node = self.rotate(node); node = self.rotate(node);
// //
return node; return node;
} }
@ -171,7 +171,7 @@ pub fn AVLTree(comptime T: type) type {
self.updateHeight(node); // self.updateHeight(node); //
// 2. 使 // 2. 使
node = self.rotate(node); node = self.rotate(node);
// //
return node; return node;
} }

View file

@ -18,7 +18,7 @@ fn hierOrder(comptime T: type, mem_allocator: std.mem.Allocator, root: *inc.Tree
while (queue.len > 0) { while (queue.len > 0) {
var queue_node = queue.popFirst().?; // var queue_node = queue.popFirst().?; //
var node = queue_node.data; var node = queue_node.data;
try list.append(node.val); // try list.append(node.val); //
if (node.left != null) { if (node.left != null) {
var tmp_node = try mem_allocator.create(L.Node); var tmp_node = try mem_allocator.create(L.Node);
tmp_node.data = node.left.?; tmp_node.data = node.left.?;

View file

@ -117,14 +117,7 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
=== "Java" === "Java"
```java title="array.java" ```java title="array.java"
/* 随机返回一个数组元素 */ [class]{array}-[func]{randomAccess}
int randomAccess(int[] nums) {
// 在区间 [0, nums.length) 中随机抽取一个数字
int randomIndex = ThreadLocalRandom.current().
nextInt(0, nums.length);
int randomNum = nums[randomIndex];
return randomNum;
}
``` ```
=== "C++" === "C++"
@ -239,17 +232,7 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
=== "Java" === "Java"
```java title="array.java" ```java title="array.java"
/* 扩展数组长度 */ [class]{array}-[func]{extend}
int[] extend(int[] nums, int enlarge) {
// 初始化一个扩展长度后的数组
int[] res = new int[nums.length + enlarge];
// 将原数组中的所有元素复制到新数组
for (int i = 0; i < nums.length; i++) {
res[i] = nums[i];
}
// 返回扩展后的新数组
return res;
}
``` ```
=== "C++" === "C++"
@ -392,23 +375,9 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
=== "Java" === "Java"
```java title="array.java" ```java title="array.java"
/* 在数组的索引 index 处插入元素 num */ [class]{array}-[func]{insert}
void insert(int[] nums, int num, int index) {
// 把索引 index 以及之后的所有元素向后移动一位
for (int i = nums.length - 1; i > index; i--) {
nums[i] = nums[i - 1];
}
// 将 num 赋给 index 处元素
nums[index] = num;
}
/* 删除索引 index 处元素 */ [class]{array}-[func]{remove}
void remove(int[] nums, int index) {
// 把索引 index 之后的所有元素向前移动一位
for (int i = index; i < nums.length - 1; i++) {
nums[i] = nums[i + 1];
}
}
``` ```
=== "C++" === "C++"
@ -592,18 +561,7 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
=== "Java" === "Java"
```java title="array.java" ```java title="array.java"
/* 遍历数组 */ [class]{array}-[func]{traverse}
void traverse(int[] nums) {
int count = 0;
// 通过索引遍历数组
for (int i = 0; i < nums.length; i++) {
count++;
}
// 直接遍历数组
for (int num : nums) {
count++;
}
}
``` ```
=== "C++" === "C++"
@ -743,14 +701,7 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
=== "Java" === "Java"
```java title="array.java" ```java title="array.java"
/* 在数组中查找指定元素 */ [class]{array}-[func]{find}
int find(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
if (nums[i] == target)
return i;
}
return -1;
}
``` ```
=== "C++" === "C++"

View file

@ -325,22 +325,9 @@ comments: true
=== "Java" === "Java"
```java title="linked_list.java" ```java title="linked_list.java"
/* 在链表的结点 n0 之后插入结点 P */ [class]{linked_list}-[func]{insert}
void insert(ListNode n0, ListNode P) {
ListNode n1 = n0.next;
n0.next = P;
P.next = n1;
}
/* 删除链表的结点 n0 之后的首个结点 */ [class]{linked_list}-[func]{remove}
void remove(ListNode n0) {
if (n0.next == null)
return;
// n0 -> P -> n1
ListNode P = n0.next;
ListNode n1 = P.next;
n0.next = n1;
}
``` ```
=== "C++" === "C++"
@ -518,15 +505,7 @@ comments: true
=== "Java" === "Java"
```java title="linked_list.java" ```java title="linked_list.java"
/* 访问链表中索引为 index 的结点 */ [class]{linked_list}-[func]{access}
ListNode access(ListNode head, int index) {
for (int i = 0; i < index; i++) {
if (head == null)
return null;
head = head.next;
}
return head;
}
``` ```
=== "C++" === "C++"
@ -655,17 +634,7 @@ comments: true
=== "Java" === "Java"
```java title="linked_list.java" ```java title="linked_list.java"
/* 在链表中查找值为 target 的首个结点 */ [class]{linked_list}-[func]{find}
int find(ListNode head, int target) {
int index = 0;
while (head != null) {
if (head.val == target)
return index;
head = head.next;
index++;
}
return -1;
}
``` ```
=== "C++" === "C++"

View file

@ -716,92 +716,7 @@ comments: true
=== "Java" === "Java"
```java title="my_list.java" ```java title="my_list.java"
/* 列表类简易实现 */ [class]{MyList}-[func]{}
class MyList {
private int[] nums; // 数组(存储列表元素)
private int capacity = 10; // 列表容量
private int size = 0; // 列表长度(即当前元素数量)
private int extendRatio = 2; // 每次列表扩容的倍数
/* 构造函数 */
public MyList() {
nums = new int[capacity];
}
/* 获取列表长度(即当前元素数量)*/
public int size() {
return size;
}
/* 获取列表容量 */
public int capacity() {
return capacity;
}
/* 访问元素 */
public int get(int index) {
// 索引如果越界则抛出异常,下同
if (index < 0 || index >= size)
throw new IndexOutOfBoundsException("索引越界");
return nums[index];
}
/* 更新元素 */
public void set(int index, int num) {
if (index < 0 || index >= size)
throw new IndexOutOfBoundsException("索引越界");
nums[index] = num;
}
/* 尾部添加元素 */
public void add(int num) {
// 元素数量超出容量时,触发扩容机制
if (size == capacity())
extendCapacity();
nums[size] = num;
// 更新元素数量
size++;
}
/* 中间插入元素 */
public void insert(int index, int num) {
if (index < 0 || index >= size)
throw new IndexOutOfBoundsException("索引越界");
// 元素数量超出容量时,触发扩容机制
if (size == capacity())
extendCapacity();
// 将索引 index 以及之后的元素都向后移动一位
for (int j = size - 1; j >= index; j--) {
nums[j + 1] = nums[j];
}
nums[index] = num;
// 更新元素数量
size++;
}
/* 删除元素 */
public int remove(int index) {
if (index < 0 || index >= size)
throw new IndexOutOfBoundsException("索引越界");
int num = nums[index];
// 将索引 index 之后的元素都向前移动一位
for (int j = index; j < size - 1; j++) {
nums[j] = nums[j + 1];
}
// 更新元素数量
size--;
// 返回被删除元素
return num;
}
/* 列表扩容 */
public void extendCapacity() {
// 新建一个长度为 size 的数组,并将原数组拷贝到新数组
nums = Arrays.copyOf(nums, capacity() * extendRatio);
// 更新列表容量
capacity = nums.length;
}
}
``` ```
=== "C++" === "C++"

View file

@ -582,22 +582,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 常数阶 */ [class]{space_complexity}-[func]{constant}
void constant(int n) {
// 常量、变量、对象占用 O(1) 空间
final int a = 0;
int b = 0;
int[] nums = new int[10000];
ListNode node = new ListNode(0);
// 循环中的变量占用 O(1) 空间
for (int i = 0; i < n; i++) {
int c = 0;
}
// 循环中的函数占用 O(1) 空间
for (int i = 0; i < n; i++) {
function();
}
}
``` ```
=== "C++" === "C++"
@ -778,21 +763,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 线性阶 */ [class]{space_complexity}-[func]{linear}
void linear(int n) {
// 长度为 n 的数组占用 O(n) 空间
int[] nums = new int[n];
// 长度为 n 的列表占用 O(n) 空间
List<ListNode> nodes = new ArrayList<>();
for (int i = 0; i < n; i++) {
nodes.add(new ListNode(i));
}
// 长度为 n 的哈希表占用 O(n) 空间
Map<Integer, String> map = new HashMap<>();
for (int i = 0; i < n; i++) {
map.put(i, String.valueOf(i));
}
}
``` ```
=== "C++" === "C++"
@ -956,12 +927,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 线性阶(递归实现) */ [class]{space_complexity}-[func]{linearRecur}
void linearRecur(int n) {
System.out.println("递归 n = " + n);
if (n == 1) return;
linearRecur(n - 1);
}
``` ```
=== "C++" === "C++"
@ -1069,20 +1035,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 平方阶 */ [class]{space_complexity}-[func]{quadratic}
void quadratic(int n) {
// 矩阵占用 O(n^2) 空间
int [][]numMatrix = new int[n][n];
// 二维列表占用 O(n^2) 空间
List<List<Integer>> numList = new ArrayList<>();
for (int i = 0; i < n; i++) {
List<Integer> tmp = new ArrayList<>();
for (int j = 0; j < n; j++) {
tmp.add(0);
}
numList.add(tmp);
}
}
``` ```
=== "C++" === "C++"
@ -1223,13 +1176,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 平方阶(递归实现) */ [class]{space_complexity}-[func]{quadraticRecur}
int quadraticRecur(int n) {
if (n <= 0) return 0;
// 数组 nums 长度为 n, n-1, ..., 2, 1
int[] nums = new int[n];
return quadraticRecur(n - 1);
}
``` ```
=== "C++" === "C++"
@ -1344,14 +1291,7 @@ $$
=== "Java" === "Java"
```java title="space_complexity.java" ```java title="space_complexity.java"
/* 指数阶(建立满二叉树) */ [class]{space_complexity}-[func]{buildTree}
TreeNode buildTree(int n) {
if (n == 0) return null;
TreeNode root = new TreeNode(0);
root.left = buildTree(n - 1);
root.right = buildTree(n - 1);
return root;
}
``` ```
=== "C++" === "C++"

View file

@ -33,19 +33,7 @@ comments: true
=== "Java" === "Java"
```java title="leetcode_two_sum.java" ```java title="leetcode_two_sum.java"
class SolutionBruteForce { [class]{SolutionBruteForce}-[func]{}
public int[] twoSum(int[] nums, int target) {
int size = nums.length;
// 两层循环,时间复杂度 O(n^2)
for (int i = 0; i < size - 1; i++) {
for (int j = i + 1; j < size; j++) {
if (nums[i] + nums[j] == target)
return new int[] { i, j };
}
}
return new int[0];
}
}
``` ```
=== "C++" === "C++"
@ -199,21 +187,7 @@ comments: true
=== "Java" === "Java"
```java title="leetcode_two_sum.java" ```java title="leetcode_two_sum.java"
class SolutionHashMap { [class]{SolutionHashMap}-[func]{}
public int[] twoSum(int[] nums, int target) {
int size = nums.length;
// 辅助哈希表,空间复杂度 O(n)
Map<Integer, Integer> dic = new HashMap<>();
// 单层循环,时间复杂度 O(n)
for (int i = 0; i < size; i++) {
if (dic.containsKey(target - nums[i])) {
return new int[] { dic.get(target - nums[i]), i };
}
dic.put(nums[i], i);
}
return new int[0];
}
}
``` ```
=== "C++" === "C++"

View file

@ -795,14 +795,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 常数阶 */ [class]{time_complexity}-[func]{constant}
int constant(int n) {
int count = 0;
int size = 100000;
for (int i = 0; i < size; i++)
count++;
return count;
}
``` ```
=== "C++" === "C++"
@ -928,13 +921,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 线性阶 */ [class]{time_complexity}-[func]{linear}
int linear(int n) {
int count = 0;
for (int i = 0; i < n; i++)
count++;
return count;
}
``` ```
=== "C++" === "C++"
@ -1052,15 +1039,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 线性阶(遍历数组) */ [class]{time_complexity}-[func]{arrayTraversal}
int arrayTraversal(int[] nums) {
int count = 0;
// 循环次数与数组长度成正比
for (int num : nums) {
count++;
}
return count;
}
``` ```
=== "C++" === "C++"
@ -1190,17 +1169,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 平方阶 */ [class]{time_complexity}-[func]{quadratic}
int quadratic(int n) {
int count = 0;
// 循环次数与数组长度成平方关系
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
count++;
}
}
return count;
}
``` ```
=== "C++" === "C++"
@ -1355,24 +1324,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 平方阶(冒泡排序) */ [class]{time_complexity}-[func]{bubbleSort}
int bubbleSort(int[] nums) {
int count = 0; // 计数器
// 外循环:待排序元素数量为 n-1, n-2, ..., 1
for (int i = nums.length - 1; i > 0; i--) {
// 内循环:冒泡操作
for (int j = 0; j < i; j++) {
if (nums[j] > nums[j + 1]) {
// 交换 nums[j] 与 nums[j + 1]
int tmp = nums[j];
nums[j] = nums[j + 1];
nums[j + 1] = tmp;
count += 3; // 元素交换包含 3 个单元操作
}
}
}
return count;
}
``` ```
=== "C++" === "C++"
@ -1585,19 +1537,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 指数阶(循环实现) */ [class]{time_complexity}-[func]{exponential}
int exponential(int n) {
int count = 0, base = 1;
// cell 每轮一分为二,形成数列 1, 2, 4, 8, ..., 2^(n-1)
for (int i = 0; i < n; i++) {
for (int j = 0; j < base; j++) {
count++;
}
base *= 2;
}
// count = 1 + 2 + 4 + 8 + .. + 2^(n-1) = 2^n - 1
return count;
}
``` ```
=== "C++" === "C++"
@ -1770,11 +1710,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 指数阶(递归实现) */ [class]{time_complexity}-[func]{expRecur}
int expRecur(int n) {
if (n == 1) return 1;
return expRecur(n - 1) + expRecur(n - 1) + 1;
}
``` ```
=== "C++" === "C++"
@ -1880,15 +1816,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 对数阶(循环实现) */ [class]{time_complexity}-[func]{logarithmic}
int logarithmic(float n) {
int count = 0;
while (n > 1) {
n = n / 2;
count++;
}
return count;
}
``` ```
=== "C++" === "C++"
@ -2024,11 +1952,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 对数阶(递归实现) */ [class]{time_complexity}-[func]{logRecur}
int logRecur(float n) {
if (n <= 1) return 0;
return logRecur(n / 2) + 1;
}
``` ```
=== "C++" === "C++"
@ -2132,16 +2056,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 线性对数阶 */ [class]{time_complexity}-[func]{linearLogRecur}
int linearLogRecur(float n) {
if (n <= 1) return 1;
int count = linearLogRecur(n / 2) +
linearLogRecur(n / 2);
for (int i = 0; i < n; i++) {
count++;
}
return count;
}
``` ```
=== "C++" === "C++"
@ -2292,16 +2207,7 @@ $$
=== "Java" === "Java"
```java title="time_complexity.java" ```java title="time_complexity.java"
/* 阶乘阶(递归实现) */ [class]{time_complexity}-[func]{factorialRecur}
int factorialRecur(int n) {
if (n == 0) return 1;
int count = 0;
// 从 1 个分裂出 n 个
for (int i = 0; i < n; i++) {
count += factorialRecur(n - 1);
}
return count;
}
``` ```
=== "C++" === "C++"
@ -2452,35 +2358,9 @@ $$
=== "Java" === "Java"
```java title="worst_best_time_complexity.java" ```java title="worst_best_time_complexity.java"
public class worst_best_time_complexity { [class]{worst_best_time_complexity}-[func]{randomNumbers}
/* 生成一个数组,元素为 { 1, 2, ..., n },顺序被打乱 */
int[] randomNumbers(int n) {
Integer[] nums = new Integer[n];
// 生成数组 nums = { 1, 2, 3, ..., n }
for (int i = 0; i < n; i++) {
nums[i] = i + 1;
}
// 随机打乱数组元素
Collections.shuffle(Arrays.asList(nums));
// Integer[] -> int[]
int[] res = new int[n];
for (int i = 0; i < n; i++) {
res[i] = nums[i];
}
return res;
}
/* 查找数组 nums 中数字 1 所在索引 */ [class]{worst_best_time_complexity}-[func]{findOne}
int findOne(int[] nums) {
for (int i = 0; i < nums.length; i++) {
// 当元素 1 在数组头部时,达到最佳时间复杂度 O(1)
// 当元素 1 在数组尾部时,达到最差时间复杂度 O(n)
if (nums[i] == 1)
return i;
}
return -1;
}
}
``` ```
=== "C++" === "C++"

View file

@ -35,83 +35,7 @@ comments: true
=== "Java" === "Java"
```java title="graph_adjacency_matrix.java" ```java title="graph_adjacency_matrix.java"
/* 基于邻接矩阵实现的无向图类 */ [class]{GraphAdjMat}-[func]{}
class GraphAdjMat {
List<Integer> vertices; // 顶点列表,元素代表“顶点值”,索引代表“顶点索引”
List<List<Integer>> adjMat; // 邻接矩阵,行列索引对应“顶点索引”
/* 构造函数 */
public GraphAdjMat(int[] vertices, int[][] edges) {
this.vertices = new ArrayList<>();
this.adjMat = new ArrayList<>();
// 添加顶点
for (int val : vertices) {
addVertex(val);
}
// 添加边
// 请注意edges 元素代表顶点索引,即对应 vertices 元素索引
for (int[] e : edges) {
addEdge(e[0], e[1]);
}
}
/* 获取顶点数量 */
public int size() {
return vertices.size();
}
/* 添加顶点 */
public void addVertex(int val) {
int n = size();
// 向顶点列表中添加新顶点的值
vertices.add(val);
// 在邻接矩阵中添加一行
List<Integer> newRow = new ArrayList<>(n);
for (int j = 0; j < n; j++) {
newRow.add(0);
}
adjMat.add(newRow);
// 在邻接矩阵中添加一列
for (List<Integer> row : adjMat) {
row.add(0);
}
}
/* 删除顶点 */
public void removeVertex(int index) {
if (index >= size())
throw new IndexOutOfBoundsException();
// 在顶点列表中移除索引 index 的顶点
vertices.remove(index);
// 在邻接矩阵中删除索引 index 的行
adjMat.remove(index);
// 在邻接矩阵中删除索引 index 的列
for (List<Integer> row : adjMat) {
row.remove(index);
}
}
/* 添加边 */
// 参数 i, j 对应 vertices 元素索引
public void addEdge(int i, int j) {
// 索引越界与相等处理
if (i < 0 || j < 0 || i >= size() || j >= size() || i == j)
throw new IndexOutOfBoundsException();
// 在无向图中,邻接矩阵沿主对角线对称,即满足 (i, j) == (j, i)
adjMat.get(i).set(j, 1);
adjMat.get(j).set(i, 1);
}
/* 删除边 */
// 参数 i, j 对应 vertices 元素索引
public void removeEdge(int i, int j) {
// 索引越界与相等处理
if (i < 0 || j < 0 || i >= size() || j >= size() || i == j)
throw new IndexOutOfBoundsException();
adjMat.get(i).set(j, 0);
adjMat.get(j).set(i, 0);
}
}
``` ```
=== "C++" === "C++"
@ -357,73 +281,9 @@ comments: true
=== "Java" === "Java"
```java title="graph_adjacency_list.java" ```java title="graph_adjacency_list.java"
/* 顶点类 */ [class]{Vertex}-[func]{}
class Vertex {
int val;
public Vertex(int val) {
this.val = val;
}
}
/* 基于邻接表实现的无向图类 */ [class]{GraphAdjList}-[func]{}
class GraphAdjList {
// 请注意vertices 和 adjList 中存储的都是 Vertex 对象
Map<Vertex, Set<Vertex>> adjList; // 邻接表(使用哈希表实现)
/* 构造函数 */
public GraphAdjList(Vertex[][] edges) {
this.adjList = new HashMap<>();
// 添加所有顶点和边
for (Vertex[] edge : edges) {
addVertex(edge[0]);
addVertex(edge[1]);
addEdge(edge[0], edge[1]);
}
}
/* 获取顶点数量 */
public int size() {
return adjList.size();
}
/* 添加边 */
public void addEdge(Vertex vet1, Vertex vet2) {
if (!adjList.containsKey(vet1) || !adjList.containsKey(vet2) || vet1 == vet2)
throw new IllegalArgumentException();
// 添加边 vet1 - vet2
adjList.get(vet1).add(vet2);
adjList.get(vet2).add(vet1);
}
/* 删除边 */
public void removeEdge(Vertex vet1, Vertex vet2) {
if (!adjList.containsKey(vet1) || !adjList.containsKey(vet2) || vet1 == vet2)
throw new IllegalArgumentException();
// 删除边 vet1 - vet2
adjList.get(vet1).remove(vet2);
adjList.get(vet2).remove(vet1);
}
/* 添加顶点 */
public void addVertex(Vertex vet) {
if (adjList.containsKey(vet))
return;
// 在邻接表中添加一个新链表(即 HashSet
adjList.put(vet, new HashSet<>());
}
/* 删除顶点 */
public void removeVertex(Vertex vet) {
if (!adjList.containsKey(vet))
throw new IllegalArgumentException();
// 在邻接表中删除顶点 vet 对应的链表(即 HashSet
adjList.remove(vet);
// 遍历其它顶点的链表(即 HashSet删除所有包含 vet 的边
for (Set<Vertex> set : adjList.values()) {
set.remove(vet);
}
}
}
``` ```
=== "C++" === "C++"

View file

@ -416,55 +416,9 @@ $$
=== "Java" === "Java"
```java title="array_hash_map.java" ```java title="array_hash_map.java"
/* 键值对 int->String */ [class]{Entry}-[func]{}
class Entry {
public int key; // 键
public String val; // 值
public Entry(int key, String val) {
this.key = key;
this.val = val;
}
}
/* 基于数组简易实现的哈希表 */ [class]{ArrayHashMap}-[func]{}
class ArrayHashMap {
private List<Entry> bucket;
public ArrayHashMap() {
// 初始化一个长度为 100 的桶(数组)
bucket = new ArrayList<>();
for (int i = 0; i < 100; i++) {
bucket.add(null);
}
}
/* 哈希函数 */
private int hashFunc(int key) {
int index = key % 100;
return index;
}
/* 查询操作 */
public String get(int key) {
int index = hashFunc(key);
Entry pair = bucket.get(index);
if (pair == null) return null;
return pair.val;
}
/* 添加操作 */
public void put(int key, String val) {
Entry pair = new Entry(key, val);
int index = hashFunc(key);
bucket.set(index, pair);
}
/* 删除操作 */
public void remove(int key) {
int index = hashFunc(key);
// 置为 null代表删除
bucket.set(index, null);
}
}
``` ```
=== "C++" === "C++"

View file

@ -263,25 +263,13 @@ comments: true
// 使用列表而非数组,这样无需考虑扩容问题 // 使用列表而非数组,这样无需考虑扩容问题
List<Integer> maxHeap; List<Integer> maxHeap;
/* 构造函数,建立空堆 */ [class]{MaxHeap}-[func]{MaxHeap}
public MaxHeap() {
maxHeap = new ArrayList<>();
}
/* 获取左子结点索引 */ [class]{MaxHeap}-[func]{left}
int left(int i) {
return 2 * i + 1;
}
/* 获取右子结点索引 */ [class]{MaxHeap}-[func]{right}
int right(int i) {
return 2 * i + 2;
}
/* 获取父结点索引 */ [class]{MaxHeap}-[func]{parent}
int parent(int i) {
return (i - 1) / 2; // 向下整除
}
``` ```
=== "C++" === "C++"
@ -407,10 +395,7 @@ comments: true
=== "Java" === "Java"
```java title="my_heap.java" ```java title="my_heap.java"
/* 访问堆顶元素 */ [class]{MaxHeap}-[func]{peek}
public int peek() {
return maxHeap.get(0);
}
``` ```
=== "C++" === "C++"
@ -505,28 +490,9 @@ comments: true
=== "Java" === "Java"
```java title="my_heap.java" ```java title="my_heap.java"
/* 元素入堆 */ [class]{MaxHeap}-[func]{push}
void push(int val) {
// 添加结点
maxHeap.add(val);
// 从底至顶堆化
siftUp(size() - 1);
}
/* 从结点 i 开始,从底至顶堆化 */ [class]{MaxHeap}-[func]{siftUp}
void siftUp(int i) {
while (true) {
// 获取结点 i 的父结点
int p = parent(i);
// 若“越过根结点”或“结点无需修复”,则结束堆化
if (p < 0 || maxHeap.get(i) <= maxHeap.get(p))
break;
// 交换两结点
swap(i, p);
// 循环向上堆化
i = p;
}
}
``` ```
=== "C++" === "C++"
@ -694,38 +660,9 @@ comments: true
=== "Java" === "Java"
```java title="my_heap.java" ```java title="my_heap.java"
/* 元素出堆 */ [class]{MaxHeap}-[func]{poll}
int poll() {
// 判空处理
if (isEmpty())
throw new EmptyStackException();
// 交换根结点与最右叶结点(即交换首元素与尾元素)
swap(0, size() - 1);
// 删除结点
int val = maxHeap.remove(size() - 1);
// 从顶至底堆化
siftDown(0);
// 返回堆顶元素
return val;
}
/* 从结点 i 开始,从顶至底堆化 */ [class]{MaxHeap}-[func]{siftDown}
void siftDown(int i) {
while (true) {
// 判断结点 i, l, r 中值最大的结点,记为 ma
int l = left(i), r = right(i), ma = i;
if (l < size() && maxHeap.get(l) > maxHeap.get(ma))
ma = l;
if (r < size() && maxHeap.get(r) > maxHeap.get(ma))
ma = r;
// 若“结点 i 最大”或“越过叶结点”,则结束堆化
if (ma == i) break;
// 交换两结点
swap(i, ma);
// 循环向下堆化
i = ma;
}
}
``` ```
=== "C++" === "C++"
@ -901,15 +838,7 @@ comments: true
=== "Java" === "Java"
```java title="my_heap.java" ```java title="my_heap.java"
/* 构造函数,根据输入列表建堆 */ [class]{MaxHeap}-[func]{MaxHeap}
public MaxHeap(List<Integer> nums) {
// 将列表元素原封不动添加进堆
maxHeap = new ArrayList<>(nums);
// 堆化除叶结点以外的其他所有结点
for (int i = parent(size() - 1); i >= 0; i--) {
siftDown(i);
}
}
``` ```
=== "C++" === "C++"

View file

@ -54,23 +54,7 @@ $$
=== "Java" === "Java"
```java title="binary_search.java" ```java title="binary_search.java"
/* 二分查找(双闭区间) */ [class]{binary_search}-[func]{binarySearch}
int binarySearch(int[] nums, int target) {
// 初始化双闭区间 [0, n-1] ,即 i, j 分别指向数组首元素、尾元素
int i = 0, j = nums.length - 1;
// 循环,当搜索区间为空时跳出(当 i > j 时为空)
while (i <= j) {
int m = (i + j) / 2; // 计算中点索引 m
if (nums[m] < target) // 此情况说明 target 在区间 [m+1, j]
i = m + 1;
else if (nums[m] > target) // 此情况说明 target 在区间 [i, m-1] 中
j = m - 1;
else // 找到目标元素,返回其索引
return m;
}
// 未找到目标元素,返回 -1
return -1;
}
``` ```
=== "C++" === "C++"
@ -235,23 +219,7 @@ $$
=== "Java" === "Java"
```java title="binary_search.java" ```java title="binary_search.java"
/* 二分查找(左闭右开) */ [class]{binary_search}-[func]{binarySearch1}
int binarySearch1(int[] nums, int target) {
// 初始化左闭右开 [0, n) ,即 i, j 分别指向数组首元素、尾元素+1
int i = 0, j = nums.length;
// 循环,当搜索区间为空时跳出(当 i = j 时为空)
while (i < j) {
int m = (i + j) / 2; // 计算中点索引 m
if (nums[m] < target) // 此情况说明 target 在区间 [m+1, j)
i = m + 1;
else if (nums[m] > target) // 此情况说明 target 在区间 [i, m) 中
j = m;
else // 找到目标元素,返回其索引
return m;
}
// 未找到目标元素,返回 -1
return -1;
}
``` ```
=== "C++" === "C++"

View file

@ -19,12 +19,7 @@ comments: true
=== "Java" === "Java"
```java title="hashing_search.java" ```java title="hashing_search.java"
/* 哈希查找(数组) */ [class]{hashing_search}-[func]{hashingSearchArray}
int hashingSearchArray(Map<Integer, Integer> map, int target) {
// 哈希表的 key: 目标元素value: 索引
// 若哈希表中无此 key ,返回 -1
return map.getOrDefault(target, -1);
}
``` ```
=== "C++" === "C++"
@ -125,12 +120,7 @@ comments: true
=== "Java" === "Java"
```java title="hashing_search.java" ```java title="hashing_search.java"
/* 哈希查找(链表) */ [class]{hashing_search}-[func]{hashingSearchLinkedList}
ListNode hashingSearchLinkedList(Map<Integer, ListNode> map, int target) {
// 哈希表的 key: 目标结点值value: 结点对象
// 若哈希表中无此 key ,返回 null
return map.getOrDefault(target, null);
}
``` ```
=== "C++" === "C++"

View file

@ -15,17 +15,7 @@ comments: true
=== "Java" === "Java"
```java title="linear_search.java" ```java title="linear_search.java"
/* 线性查找(数组) */ [class]{hashing_search}-[func]{linearSearchArray}
int linearSearchArray(int[] nums, int target) {
// 遍历数组
for (int i = 0; i < nums.length; i++) {
// 找到目标元素,返回其索引
if (nums[i] == target)
return i;
}
// 未找到目标元素,返回 -1
return -1;
}
``` ```
=== "C++" === "C++"
@ -155,18 +145,7 @@ comments: true
=== "Java" === "Java"
```java title="linear_search.java" ```java title="linear_search.java"
/* 线性查找(链表) */ [class]{hashing_search}-[func]{linearSearchLinkedList}
ListNode linearSearchLinkedList(ListNode head, int target) {
// 遍历链表
while (head != null) {
// 找到目标结点,返回之
if (head.val == target)
return head;
head = head.next;
}
// 未找到目标结点,返回 null
return null;
}
``` ```
=== "C++" === "C++"

View file

@ -50,21 +50,7 @@ comments: true
=== "Java" === "Java"
```java title="bubble_sort.java" ```java title="bubble_sort.java"
/* 冒泡排序 */ [class]{bubble_sort}-[func]{bubbleSort}
void bubbleSort(int[] nums) {
// 外循环:待排序元素数量为 n-1, n-2, ..., 1
for (int i = nums.length - 1; i > 0; i--) {
// 内循环:冒泡操作
for (int j = 0; j < i; j++) {
if (nums[j] > nums[j + 1]) {
// 交换 nums[j] 与 nums[j + 1]
int tmp = nums[j];
nums[j] = nums[j + 1];
nums[j + 1] = tmp;
}
}
}
}
``` ```
=== "C++" === "C++"
@ -243,24 +229,7 @@ comments: true
=== "Java" === "Java"
```java title="bubble_sort.java" ```java title="bubble_sort.java"
/* 冒泡排序(标志优化)*/ [class]{bubble_sort}-[func]{bubbleSortWithFlag}
void bubbleSortWithFlag(int[] nums) {
// 外循环:待排序元素数量为 n-1, n-2, ..., 1
for (int i = nums.length - 1; i > 0; i--) {
boolean flag = false; // 初始化标志位
// 内循环:冒泡操作
for (int j = 0; j < i; j++) {
if (nums[j] > nums[j + 1]) {
// 交换 nums[j] 与 nums[j + 1]
int tmp = nums[j];
nums[j] = nums[j + 1];
nums[j + 1] = tmp;
flag = true; // 记录交换元素
}
}
if (!flag) break; // 此轮冒泡未交换任何元素,直接跳出
}
}
``` ```
=== "C++" === "C++"

View file

@ -27,19 +27,7 @@ comments: true
=== "Java" === "Java"
```java title="insertion_sort.java" ```java title="insertion_sort.java"
/* 插入排序 */ [class]{insertion_sort}-[func]{insertionSort}
void insertionSort(int[] nums) {
// 外循环base = nums[1], nums[2], ..., nums[n-1]
for (int i = 1; i < nums.length; i++) {
int base = nums[i], j = i - 1;
// 内循环:将 base 插入到左边的正确位置
while (j >= 0 && nums[j] > base) {
nums[j + 1] = nums[j]; // 1. 将 nums[j] 向右移动一位
j--;
}
nums[j + 1] = base; // 2. 将 base 赋值到正确位置
}
}
``` ```
=== "C++" === "C++"

View file

@ -62,55 +62,17 @@ comments: true
=== "Java" === "Java"
```java title="merge_sort.java" ```java title="merge_sort.java"
/** [class]{merge_sort}-[func]{merge}
* 合并左子数组和右子数组
* 左子数组区间 [left, mid]
* 右子数组区间 [mid + 1, right]
*/
void merge(int[] nums, int left, int mid, int right) {
// 初始化辅助数组
int[] tmp = Arrays.copyOfRange(nums, left, right + 1);
// 左子数组的起始索引和结束索引
int leftStart = left - left, leftEnd = mid - left;
// 右子数组的起始索引和结束索引
int rightStart = mid + 1 - left, rightEnd = right - left;
// i, j 分别指向左子数组、右子数组的首元素
int i = leftStart, j = rightStart;
// 通过覆盖原数组 nums 来合并左子数组和右子数组
for (int k = left; k <= right; k++) {
// 若“左子数组已全部合并完”,则选取右子数组元素,并且 j++
if (i > leftEnd)
nums[k] = tmp[j++];
// 否则,若“右子数组已全部合并完”或“左子数组元素 <= 右子数组元素”,则选取左子数组元素,并且 i++
else if (j > rightEnd || tmp[i] <= tmp[j])
nums[k] = tmp[i++];
// 否则,若“左右子数组都未全部合并完”且“左子数组元素 > 右子数组元素”,则选取右子数组元素,并且 j++
else
nums[k] = tmp[j++];
}
}
/* 归并排序 */ [class]{merge_sort}-[func]{mergeSort}
void mergeSort(int[] nums, int left, int right) {
// 终止条件
if (left >= right) return; // 当子数组长度为 1 时终止递归
// 递归划分
int mid = (left + right) / 2; // 计算数组中点
mergeSort(nums, left, mid); // 递归左子数组
mergeSort(nums, mid + 1, right); // 递归右子数组
// 回溯合并
merge(nums, left, mid, right);
}
``` ```
=== "C++" === "C++"
```cpp title="merge_sort.cpp" ```cpp title="merge_sort.cpp"
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
void merge(vector<int>& nums, int left, int mid, int right) { void merge(vector<int>& nums, int left, int mid, int right) {
// 初始化辅助数组 // 初始化辅助数组
vector<int> tmp(nums.begin() + left, nums.begin() + right + 1); vector<int> tmp(nums.begin() + left, nums.begin() + right + 1);
@ -210,11 +172,9 @@ comments: true
=== "JavaScript" === "JavaScript"
```js title="merge_sort.js" ```js title="merge_sort.js"
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
function merge(nums, left, mid, right) { function merge(nums, left, mid, right) {
// 初始化辅助数组 // 初始化辅助数组
let tmp = nums.slice(left, right + 1); let tmp = nums.slice(left, right + 1);
@ -255,11 +215,9 @@ comments: true
=== "TypeScript" === "TypeScript"
```typescript title="merge_sort.ts" ```typescript title="merge_sort.ts"
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
function merge(nums: number[], left: number, mid: number, right: number): void { function merge(nums: number[], left: number, mid: number, right: number): void {
// 初始化辅助数组 // 初始化辅助数组
let tmp = nums.slice(left, right + 1); let tmp = nums.slice(left, right + 1);
@ -306,11 +264,9 @@ comments: true
=== "C#" === "C#"
```csharp title="merge_sort.cs" ```csharp title="merge_sort.cs"
/** /* 合并左子数组和右子数组 */
* 合并左子数组和右子数组 // 左子数组区间 [left, mid]
* 左子数组区间 [left, mid] // 右子数组区间 [mid + 1, right]
* 右子数组区间 [mid + 1, right]
*/
void merge(int[] nums, int left, int mid, int right) void merge(int[] nums, int left, int mid, int right)
{ {
// 初始化辅助数组 // 初始化辅助数组

View file

@ -45,28 +45,10 @@ comments: true
=== "Java" === "Java"
``` java title="quick_sort.java" ```java title="quick_sort.java"
/* 元素交换 */ [class]{QuickSort}-[func]{swap}
void swap(int[] nums, int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵划分 */ [class]{QuickSort}-[func]{partition}
int partition(int[] nums, int left, int right) {
// 以 nums[left] 作为基准数
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 从右向左找首个小于基准数的元素
while (i < j && nums[i] <= nums[left])
i++; // 从左向右找首个大于基准数的元素
swap(nums, i, j); // 交换这两个元素
}
swap(nums, i, left); // 将基准数交换至两子数组的分界线
return i; // 返回基准数的索引
}
``` ```
=== "C++" === "C++"
@ -271,17 +253,7 @@ comments: true
=== "Java" === "Java"
```java title="quick_sort.java" ```java title="quick_sort.java"
/* 快速排序 */ [class]{QuickSort}-[func]{quickSort}
void quickSort(int[] nums, int left, int right) {
// 子数组长度为 1 时终止递归
if (left >= right)
return;
// 哨兵划分
int pivot = partition(nums, left, right);
// 递归左子数组、右子数组
quickSort(nums, left, pivot - 1);
quickSort(nums, pivot + 1, right);
}
``` ```
=== "C++" === "C++"
@ -435,27 +407,9 @@ comments: true
=== "Java" === "Java"
```java title="quick_sort.java" ```java title="quick_sort.java"
/* 选取三个元素的中位数 */ [class]{QuickSortMedian}-[func]{medianThree}
int medianThree(int[] nums, int left, int mid, int right) {
// 使用了异或操作来简化代码
// 异或规则为 0 ^ 0 = 1 ^ 1 = 0, 0 ^ 1 = 1 ^ 0 = 1
if ((nums[left] < nums[mid]) ^ (nums[left] < nums[right]))
return left;
else if ((nums[mid] < nums[left]) ^ (nums[mid] < nums[right]))
return mid;
else
return right;
}
/* 哨兵划分(三数取中值) */ [class]{QuickSortMedian}-[func]{partition}
int partition(int[] nums, int left, int right) {
// 选取三个候选元素的中位数
int med = medianThree(nums, left, (left + right) / 2, right);
// 将中位数交换至数组最左端
swap(nums, left, med);
// 以 nums[left] 作为基准数
// 下同省略...
}
``` ```
=== "C++" === "C++"
@ -642,22 +596,7 @@ comments: true
=== "Java" === "Java"
```java title="quick_sort.java" ```java title="quick_sort.java"
/* 快速排序(尾递归优化) */ [class]{QuickSortTailCall}-[func]{quickSort}
void quickSort(int[] nums, int left, int right) {
// 子数组长度为 1 时终止
while (left < right) {
// 哨兵划分操作
int pivot = partition(nums, left, right);
// 对两个子数组中较短的那个执行快排
if (pivot - left < right - pivot) {
quickSort(nums, left, pivot - 1); // 递归排序左子数组
left = pivot + 1; // 剩余待排序区间为 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, right); // 递归排序右子数组
right = pivot - 1; // 剩余待排序区间为 [left, pivot - 1]
}
}
}
``` ```
=== "C++" === "C++"

View file

@ -321,119 +321,9 @@ comments: true
=== "Java" === "Java"
```java title="linkedlist_deque.java" ```java title="linkedlist_deque.java"
/* 双向链表结点 */ [class]{ListNode}-[func]{}
class ListNode {
int val; // 结点值
ListNode next; // 后继结点引用(指针)
ListNode prev; // 前驱结点引用(指针)
ListNode(int val) {
this.val = val;
prev = next = null;
}
}
/* 基于双向链表实现的双向队列 */ [class]{LinkedListDeque}-[func]{}
class LinkedListDeque {
private ListNode front, rear; // 头结点 front ,尾结点 rear
private int size = 0; // 双向队列的长度
public LinkedListDeque() {
front = rear = null;
}
/* 获取双向队列的长度 */
public int size() {
return size;
}
/* 判断双向队列是否为空 */
public boolean isEmpty() {
return size() == 0;
}
/* 入队操作 */
private void push(int num, boolean isFront) {
ListNode node = new ListNode(num);
// 若链表为空,则令 front, rear 都指向 node
if (isEmpty())
front = rear = node;
// 队首入队操作
else if (isFront) {
// 将 node 添加至链表头部
front.prev = node;
node.next = front;
front = node; // 更新头结点
// 队尾入队操作
} else {
// 将 node 添加至链表尾部
rear.next = node;
node.prev = rear;
rear = node; // 更新尾结点
}
size++; // 更新队列长度
}
/* 队首入队 */
public void pushFirst(int num) {
push(num, true);
}
/* 队尾入队 */
public void pushLast(int num) {
push(num, false);
}
/* 出队操作 */
private Integer poll(boolean isFront) {
// 若队列为空,直接返回 null
if (isEmpty())
return null;
int val;
// 队首出队操作
if (isFront) {
val = front.val; // 暂存头结点值
// 删除头结点
ListNode fNext = front.next;
if (fNext != null) {
fNext.prev = null;
front.next = null;
}
front = fNext; // 更新头结点
// 队尾出队操作
} else {
val = rear.val; // 暂存尾结点值
// 删除尾结点
ListNode rPrev = rear.prev;
if (rPrev != null) {
rPrev.next = null;
rear.prev = null;
}
rear = rPrev; // 更新尾结点
}
size--; // 更新队列长度
return val;
}
/* 队首出队 */
public Integer pollFirst() {
return poll(true);
}
/* 队尾出队 */
public Integer pollLast() {
return poll(false);
}
/* 访问队首元素 */
public Integer peekFirst() {
return isEmpty() ? null : front.val;
}
/* 访问队尾元素 */
public Integer peekLast() {
return isEmpty() ? null : rear.val;
}
}
``` ```
=== "C++" === "C++"

View file

@ -284,53 +284,7 @@ comments: true
=== "Java" === "Java"
```java title="linkedlist_queue.java" ```java title="linkedlist_queue.java"
/* 基于链表实现的队列 */ [class]{LinkedListQueue}-[func]{}
class LinkedListQueue {
private ListNode front, rear; // 头结点 front ,尾结点 rear
private int queSize = 0;
public LinkedListQueue() {
front = null;
rear = null;
}
/* 获取队列的长度 */
public int size() {
return queSize;
}
/* 判断队列是否为空 */
public boolean isEmpty() {
return size() == 0;
}
/* 入队 */
public void push(int num) {
// 尾结点后添加 num
ListNode node = new ListNode(num);
// 如果队列为空,则令头、尾结点都指向该结点
if (front == null) {
front = node;
rear = node;
// 如果队列不为空,则将该结点添加到尾结点后
} else {
rear.next = node;
rear = node;
}
queSize++;
}
/* 出队 */
public int poll() {
int num = peek();
// 删除头结点
front = front.next;
queSize--;
return num;
}
/* 访问队首元素 */
public int peek() {
if (size() == 0)
throw new EmptyStackException();
return front.val;
}
}
``` ```
=== "C++" === "C++"
@ -709,62 +663,7 @@ comments: true
=== "Java" === "Java"
```java title="array_queue.java" ```java title="array_queue.java"
/* 基于环形数组实现的队列 */ [class]{ArrayQueue}-[func]{}
class ArrayQueue {
private int[] nums; // 用于存储队列元素的数组
private int front; // 队首指针,指向队首元素
private int queSize; // 队列长度
public ArrayQueue(int capacity) {
nums = new int[capacity];
front = queSize = 0;
}
/* 获取队列的容量 */
public int capacity() {
return nums.length;
}
/* 获取队列的长度 */
public int size() {
return queSize;
}
/* 判断队列是否为空 */
public boolean isEmpty() {
return queSize == 0;
}
/* 入队 */
public void push(int num) {
if (queSize == capacity()) {
System.out.println("队列已满");
return;
}
// 计算尾指针,指向队尾索引 + 1
// 通过取余操作,实现 rear 越过数组尾部后回到头部
int rear = (front + queSize) % capacity();
// 尾结点后添加 num
nums[rear] = num;
queSize++;
}
/* 出队 */
public int poll() {
int num = peek();
// 队首指针向后移动一位,若越过尾部则返回到数组头部
front = (front + 1) % capacity();
queSize--;
return num;
}
/* 访问队首元素 */
public int peek() {
if (isEmpty())
throw new EmptyStackException();
return nums[front];
}
}
``` ```
=== "C++" === "C++"

View file

@ -287,42 +287,7 @@ comments: true
=== "Java" === "Java"
```java title="linkedlist_stack.java" ```java title="linkedlist_stack.java"
/* 基于链表实现的栈 */ [class]{LinkedListStack}-[func]{}
class LinkedListStack {
private ListNode stackPeek; // 将头结点作为栈顶
private int stkSize = 0; // 栈的长度
public LinkedListStack() {
stackPeek = null;
}
/* 获取栈的长度 */
public int size() {
return stkSize;
}
/* 判断栈是否为空 */
public boolean isEmpty() {
return size() == 0;
}
/* 入栈 */
public void push(int num) {
ListNode node = new ListNode(num);
node.next = stackPeek;
stackPeek = node;
stkSize++;
}
/* 出栈 */
public int pop() {
int num = peek();
stackPeek = stackPeek.next;
stkSize--;
return num;
}
/* 访问栈顶元素 */
public int peek() {
if (size() == 0)
throw new EmptyStackException();
return stackPeek.val;
}
}
``` ```
=== "C++" === "C++"
@ -681,38 +646,7 @@ comments: true
=== "Java" === "Java"
```java title="array_stack.java" ```java title="array_stack.java"
/* 基于数组实现的栈 */ [class]{ArrayStack}-[func]{}
class ArrayStack {
private ArrayList<Integer> stack;
public ArrayStack() {
// 初始化列表(动态数组)
stack = new ArrayList<>();
}
/* 获取栈的长度 */
public int size() {
return stack.size();
}
/* 判断栈是否为空 */
public boolean isEmpty() {
return size() == 0;
}
/* 入栈 */
public void push(int num) {
stack.add(num);
}
/* 出栈 */
public int pop() {
if (isEmpty())
throw new EmptyStackException();
return stack.remove(size() - 1);
}
/* 访问栈顶元素 */
public int peek() {
if (isEmpty())
throw new EmptyStackException();
return stack.get(size() - 1);
}
}
``` ```
=== "C++" === "C++"

View file

@ -28,7 +28,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Java" === "Java"
```java title="avl_tree.java" ```java title=""
/* AVL 树结点类 */ /* AVL 树结点类 */
class TreeNode { class TreeNode {
public int val; // 结点值 public int val; // 结点值
@ -41,7 +41,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "C++" === "C++"
```cpp title="avl_tree.cpp" ```cpp title=""
/* AVL 树结点类 */ /* AVL 树结点类 */
struct TreeNode { struct TreeNode {
int val{}; // 结点值 int val{}; // 结点值
@ -55,7 +55,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Python" === "Python"
```python title="avl_tree.py" ```python title=""
""" AVL 树结点类 """ """ AVL 树结点类 """
class TreeNode: class TreeNode:
def __init__(self, val=None, left=None, right=None): def __init__(self, val=None, left=None, right=None):
@ -67,7 +67,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Go" === "Go"
```go title="avl_tree.go" ```go title=""
/* AVL 树结点类 */ /* AVL 树结点类 */
type TreeNode struct { type TreeNode struct {
Val int // 结点值 Val int // 结点值
@ -79,7 +79,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "JavaScript" === "JavaScript"
```js title="avl_tree.js" ```js title=""
class TreeNode { class TreeNode {
val; // 结点值 val; // 结点值
height; //结点高度 height; //结点高度
@ -96,7 +96,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "TypeScript" === "TypeScript"
```typescript title="avl_tree.ts" ```typescript title=""
class TreeNode { class TreeNode {
val: number; // 结点值 val: number; // 结点值
height: number; // 结点高度 height: number; // 结点高度
@ -113,13 +113,13 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "C" === "C"
```c title="avl_tree.c" ```c title=""
``` ```
=== "C#" === "C#"
```csharp title="avl_tree.cs" ```csharp title=""
/* AVL 树结点类 */ /* AVL 树结点类 */
class TreeNode { class TreeNode {
public int val; // 结点值 public int val; // 结点值
@ -132,7 +132,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Swift" === "Swift"
```swift title="avl_tree.swift" ```swift title=""
/* AVL 树结点类 */ /* AVL 树结点类 */
class TreeNode { class TreeNode {
var val: Int // 结点值 var val: Int // 结点值
@ -149,7 +149,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Zig" === "Zig"
```zig title="avl_tree.zig" ```zig title=""
``` ```
@ -158,17 +158,9 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 获取结点高度 */ [class]{AVLTree}-[func]{height}
int height(TreeNode node) {
// 空结点高度为 -1 ,叶结点高度为 0
return node == null ? -1 : node.height;
}
/* 更新结点高度 */ [class]{AVLTree}-[func]{updateHeight}
void updateHeight(TreeNode node) {
// 结点高度等于最高子树高度 + 1
node.height = Math.max(height(node.left), height(node.right)) + 1;
}
``` ```
=== "C++" === "C++"
@ -305,13 +297,7 @@ G. M. Adelson-Velsky 和 E. M. Landis 在其 1962 年发表的论文 "An algorit
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 获取结点平衡因子 */ [class]{AVLTree}-[func]{balanceFactor}
public int balanceFactor(TreeNode node) {
// 空结点平衡因子为 0
if (node == null) return 0;
// 结点平衡因子 = 左子树高度 - 右子树高度
return height(node.left) - height(node.right);
}
``` ```
=== "C++" === "C++"
@ -421,7 +407,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
### Case 1 - 右旋 ### Case 1 - 右旋
如下图所示(结点下方为「平衡因子」),从底至顶看,二叉树中首个失衡结点是 **结点 3**。我们聚焦在以该失衡结点为根结点的子树上,将该结点记为 `node` ,将其左子点记为 `child` ,执行「右旋」操作。完成右旋后,该子树已经恢复平衡,并且仍然为二叉搜索树。 如下图所示(结点下方为「平衡因子」),从底至顶看,二叉树中首个失衡结点是 **结点 3**。我们聚焦在以该失衡结点为根结点的子树上,将该结点记为 `node` ,将其左子点记为 `child` ,执行「右旋」操作。完成右旋后,该子树已经恢复平衡,并且仍然为二叉搜索树。
=== "Step 1" === "Step 1"
![right_rotate_step1](avl_tree.assets/right_rotate_step1.png) ![right_rotate_step1](avl_tree.assets/right_rotate_step1.png)
@ -444,19 +430,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 右旋操作 */ [class]{AVLTree}-[func]{rightRotate}
TreeNode rightRotate(TreeNode node) {
TreeNode child = node.left;
TreeNode grandChild = child.right;
// 以 child 为原点,将 node 向右旋转
child.right = node;
node.left = grandChild;
// 更新结点高度
updateHeight(node);
updateHeight(child);
// 返回旋转后子树的根节点
return child;
}
``` ```
=== "C++" === "C++"
@ -472,7 +446,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -496,7 +470,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node) updateHeight(node)
updateHeight(child) updateHeight(child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
``` ```
@ -514,7 +488,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -532,7 +506,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -557,7 +531,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -575,7 +549,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node: node) updateHeight(node: node)
updateHeight(node: child) updateHeight(node: child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
``` ```
@ -601,19 +575,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 左旋操作 */ [class]{AVLTree}-[func]{leftRotate}
private TreeNode leftRotate(TreeNode node) {
TreeNode child = node.right;
TreeNode grandChild = child.left;
// 以 child 为原点,将 node 向左旋转
child.left = node;
node.right = grandChild;
// 更新结点高度
updateHeight(node);
updateHeight(child);
// 返回旋转后子树的根节点
return child;
}
``` ```
=== "C++" === "C++"
@ -629,7 +591,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -653,7 +615,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node) updateHeight(node)
updateHeight(child) updateHeight(child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
``` ```
@ -671,7 +633,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -689,7 +651,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
this.updateHeight(node); this.updateHeight(node);
this.updateHeight(child); this.updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -716,7 +678,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node); updateHeight(node);
updateHeight(child); updateHeight(child);
// 返回旋转后子树的根 // 返回旋转后子树的根
return child; return child;
} }
``` ```
@ -734,7 +696,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
// 更新结点高度 // 更新结点高度
updateHeight(node: node) updateHeight(node: node)
updateHeight(node: child) updateHeight(node: child)
// 返回旋转后子树的根 // 返回旋转后子树的根
return child return child
} }
@ -782,35 +744,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 执行旋转操作,使该子树重新恢复平衡 */ [class]{AVLTree}-[func]{rotate}
TreeNode rotate(TreeNode node) {
// 获取结点 node 的平衡因子
int balanceFactor = balanceFactor(node);
// 左偏树
if (balanceFactor > 1) {
if (balanceFactor(node.left) >= 0) {
// 右旋
return rightRotate(node);
} else {
// 先左旋后右旋
node.left = leftRotate(node.left);
return rightRotate(node);
}
}
// 右偏树
if (balanceFactor < -1) {
if (balanceFactor(node.right) <= 0) {
// 左旋
return leftRotate(node);
} else {
// 先右旋后左旋
node.right = rightRotate(node.right);
return leftRotate(node);
}
}
// 平衡树,无需旋转,直接返回
return node;
}
``` ```
=== "C++" === "C++"
@ -1054,28 +988,9 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 插入结点 */ [class]{AVLTree}-[func]{insert}
TreeNode insert(int val) {
root = insertHelper(root, val);
return root;
}
/* 递归插入结点(辅助函数) */ [class]{AVLTree}-[func]{insertHelper}
TreeNode insertHelper(TreeNode node, int val) {
if (node == null) return new TreeNode(val);
/* 1. 查找插入位置,并插入结点 */
if (val < node.val)
node.left = insertHelper(node.left, val);
else if (val > node.val)
node.right = insertHelper(node.right, val);
else
return node; // 重复结点不插入,直接返回
updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node);
// 返回子树的根节点
return node;
}
``` ```
=== "C++" === "C++"
@ -1100,7 +1015,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1139,7 +1054,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node) updateHeight(node)
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node) node = rotate(node)
// 返回子树的根 // 返回子树的根
return node return node
} }
``` ```
@ -1163,7 +1078,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1191,7 +1106,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1228,7 +1143,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1260,7 +1175,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node: node) // 更新结点高度 updateHeight(node: node) // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node: node) node = rotate(node: node)
// 返回子树的根 // 返回子树的根
return node return node
} }
``` ```
@ -1278,42 +1193,11 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
=== "Java" === "Java"
```java title="avl_tree.java" ```java title="avl_tree.java"
/* 删除结点 */ [class]{AVLTree}-[func]{remove}
TreeNode remove(int val) {
root = removeHelper(root, val);
return root;
}
/* 递归删除结点(辅助函数) */ [class]{AVLTree}-[func]{removeHelper}
TreeNode removeHelper(TreeNode node, int val) {
if (node == null) return null; [class]{AVLTree}-[func]{getInOrderNext}
/* 1. 查找结点,并删除之 */
if (val < node.val)
node.left = removeHelper(node.left, val);
else if (val > node.val)
node.right = removeHelper(node.right, val);
else {
if (node.left == null || node.right == null) {
TreeNode child = node.left != null ? node.left : node.right;
// 子结点数量 = 0 ,直接删除 node 并返回
if (child == null)
return null;
// 子结点数量 = 1 ,直接删除 node
else
node = child;
} else {
// 子结点数量 = 2 ,则将中序遍历的下个结点删除,并用该结点替换当前结点
TreeNode temp = getInOrderNext(node.right);
node.right = removeHelper(node.right, temp.val);
node.val = temp.val;
}
}
updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node);
// 返回子树的根节点
return node;
}
``` ```
=== "C++" === "C++"
@ -1356,7 +1240,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1412,7 +1296,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node) updateHeight(node)
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node) node = rotate(node)
// 返回子树的根 // 返回子树的根
return node return node
} }
``` ```
@ -1449,7 +1333,17 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根节点 // 返回子树的根结点
return node;
}
/* 获取中序遍历中的下一个结点(仅适用于 root 有左子结点的情况) */
getInOrderNext(node) {
if (node === null) return node;
// 循环访问左子结点,直到叶结点时为最小结点,跳出
while (node.left !== null) {
node = node.left;
}
return node; return node;
} }
@ -1501,7 +1395,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
this.updateHeight(node); // 更新结点高度 this.updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = this.rotate(node); node = this.rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
@ -1564,7 +1458,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node); // 更新结点高度 updateHeight(node); // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node); node = rotate(node);
// 返回子树的根 // 返回子树的根
return node; return node;
} }
``` ```
@ -1611,7 +1505,7 @@ AVL 树的独特之处在于「旋转 Rotation」的操作其可 **在不影
updateHeight(node: node) // 更新结点高度 updateHeight(node: node) // 更新结点高度
/* 2. 执行旋转操作,使该子树重新恢复平衡 */ /* 2. 执行旋转操作,使该子树重新恢复平衡 */
node = rotate(node: node) node = rotate(node: node)
// 返回子树的根 // 返回子树的根
return node return node
} }
``` ```

View file

@ -38,21 +38,7 @@ comments: true
=== "Java" === "Java"
```java title="binary_search_tree.java" ```java title="binary_search_tree.java"
/* 查找结点 */ [class]{BinarySearchTree}-[func]{search}
TreeNode search(int num) {
TreeNode cur = root;
// 循环查找,越过叶结点后跳出
while (cur != null) {
// 目标结点在 cur 的右子树中
if (cur.val < num) cur = cur.right;
// 目标结点在 cur 的左子树中
else if (cur.val > num) cur = cur.left;
// 找到目标结点,跳出循环
else break;
}
// 返回目标结点
return cur;
}
``` ```
=== "C++" === "C++"
@ -220,27 +206,7 @@ comments: true
=== "Java" === "Java"
```java title="binary_search_tree.java" ```java title="binary_search_tree.java"
/* 插入结点 */ [class]{BinarySearchTree}-[func]{insert}
TreeNode insert(int num) {
// 若树为空,直接提前返回
if (root == null) return null;
TreeNode cur = root, pre = null;
// 循环查找,越过叶结点后跳出
while (cur != null) {
// 找到重复结点,直接返回
if (cur.val == num) return null;
pre = cur;
// 插入位置在 cur 的右子树中
if (cur.val < num) cur = cur.right;
// 插入位置在 cur 的左子树中
else cur = cur.left;
}
// 插入结点 val
TreeNode node = new TreeNode(num);
if (pre.val < num) pre.right = node;
else pre.left = node;
return node;
}
``` ```
=== "C++" === "C++"
@ -491,55 +457,9 @@ comments: true
=== "Java" === "Java"
```java title="binary_search_tree.java" ```java title="binary_search_tree.java"
/* 删除结点 */ [class]{BinarySearchTree}-[func]{remove}
TreeNode remove(int num) {
// 若树为空,直接提前返回
if (root == null) return null;
TreeNode cur = root, pre = null;
// 循环查找,越过叶结点后跳出
while (cur != null) {
// 找到待删除结点,跳出循环
if (cur.val == num) break;
pre = cur;
// 待删除结点在 cur 的右子树中
if (cur.val < num) cur = cur.right;
// 待删除结点在 cur 的左子树中
else cur = cur.left;
}
// 若无待删除结点,则直接返回
if (cur == null) return null;
// 子结点数量 = 0 or 1
if (cur.left == null || cur.right == null) {
// 当子结点数量 = 0 / 1 时, child = null / 该子结点
TreeNode child = cur.left != null ? cur.left : cur.right;
// 删除结点 cur
if (pre.left == cur) pre.left = child;
else pre.right = child;
// 释放内存
delete cur;
}
// 子结点数量 = 2
else {
// 获取中序遍历中 cur 的下一个结点
TreeNode nex = getInOrderNext(cur.right);
int tmp = nex.val;
// 递归删除结点 nex
remove(nex.val);
// 将 nex 的值复制给 cur
cur.val = tmp;
}
return cur;
}
/* 获取中序遍历中的下一个结点(仅适用于 root 有左子结点的情况) */ [class]{BinarySearchTree}-[func]{getInOrderNext}
public TreeNode getInOrderNext(TreeNode root) {
if (root == null) return root;
// 循环访问左子结点,直到叶结点时为最小结点,跳出
while (root.left != null) {
root = root.left;
}
return root;
}
``` ```
=== "C++" === "C++"

View file

@ -197,7 +197,7 @@ comments: true
```python title="binary_tree.py" ```python title="binary_tree.py"
""" 初始化二叉树 """ """ 初始化二叉树 """
# 初始化 # 初始化
n1 = TreeNode(val=1) n1 = TreeNode(val=1)
n2 = TreeNode(val=2) n2 = TreeNode(val=2)
n3 = TreeNode(val=3) n3 = TreeNode(val=3)
@ -343,7 +343,7 @@ comments: true
# 在 n1 -> n2 中间插入结点 P # 在 n1 -> n2 中间插入结点 P
n1.left = p n1.left = p
p.left = n2 p.left = n2
# 删除点 P # 删除点 P
n1.left = n2 n1.left = n2
``` ```

View file

@ -21,22 +21,7 @@ comments: true
=== "Java" === "Java"
```java title="binary_tree_bfs.java" ```java title="binary_tree_bfs.java"
/* 层序遍历 */ [class]{binary_tree_bfs}-[func]{hierOrder}
List<Integer> hierOrder(TreeNode root) {
// 初始化队列,加入根结点
Queue<TreeNode> queue = new LinkedList<>() {{ add(root); }};
// 初始化一个列表,用于保存遍历序列
List<Integer> list = new ArrayList<>();
while (!queue.isEmpty()) {
TreeNode node = queue.poll(); // 队列出队
list.add(node.val); // 保存结点值
if (node.left != null)
queue.offer(node.left); // 左子结点入队
if (node.right != null)
queue.offer(node.right); // 右子结点入队
}
return list;
}
``` ```
=== "C++" === "C++"
@ -52,7 +37,7 @@ comments: true
while (!queue.empty()) { while (!queue.empty()) {
TreeNode* node = queue.front(); TreeNode* node = queue.front();
queue.pop(); // 队列出队 queue.pop(); // 队列出队
vec.push_back(node->val); // 保存结点 vec.push_back(node->val); // 保存结点
if (node->left != nullptr) if (node->left != nullptr)
queue.push(node->left); // 左子结点入队 queue.push(node->left); // 左子结点入队
if (node->right != nullptr) if (node->right != nullptr)
@ -81,7 +66,7 @@ comments: true
for queue.Len() > 0 { for queue.Len() > 0 {
// poll // poll
node := queue.Remove(queue.Front()).(*TreeNode) node := queue.Remove(queue.Front()).(*TreeNode)
// 保存结点 // 保存结点
nums = append(nums, node.Val) nums = append(nums, node.Val)
if node.Left != nil { if node.Left != nil {
// 左子结点入队 // 左子结点入队
@ -107,7 +92,7 @@ comments: true
let list = []; let list = [];
while (queue.length) { while (queue.length) {
let node = queue.shift(); // 队列出队 let node = queue.shift(); // 队列出队
list.push(node.val); // 保存结点 list.push(node.val); // 保存结点
if (node.left) if (node.left)
queue.push(node.left); // 左子结点入队 queue.push(node.left); // 左子结点入队
if (node.right) if (node.right)
@ -128,7 +113,7 @@ comments: true
const list: number[] = []; const list: number[] = [];
while (queue.length) { while (queue.length) {
let node = queue.shift() as TreeNode; // 队列出队 let node = queue.shift() as TreeNode; // 队列出队
list.push(node.val); // 保存结点 list.push(node.val); // 保存结点
if (node.left) { if (node.left) {
queue.push(node.left); // 左子结点入队 queue.push(node.left); // 左子结点入队
} }
@ -182,7 +167,7 @@ comments: true
var list: [Int] = [] var list: [Int] = []
while !queue.isEmpty { while !queue.isEmpty {
let node = queue.removeFirst() // 队列出队 let node = queue.removeFirst() // 队列出队
list.append(node.val) // 保存结点 list.append(node.val) // 保存结点
if let left = node.left { if let left = node.left {
queue.append(left) // 左子结点入队 queue.append(left) // 左子结点入队
} }
@ -223,32 +208,11 @@ comments: true
=== "Java" === "Java"
```java title="binary_tree_dfs.java" ```java title="binary_tree_dfs.java"
/* 前序遍历 */ [class]{binary_tree_dfs}-[func]{preOrder}
void preOrder(TreeNode root) {
if (root == null) return;
// 访问优先级:根结点 -> 左子树 -> 右子树
list.add(root.val);
preOrder(root.left);
preOrder(root.right);
}
/* 中序遍历 */ [class]{binary_tree_dfs}-[func]{inOrder}
void inOrder(TreeNode root) {
if (root == null) return;
// 访问优先级:左子树 -> 根结点 -> 右子树
inOrder(root.left);
list.add(root.val);
inOrder(root.right);
}
/* 后序遍历 */ [class]{binary_tree_dfs}-[func]{postOrder}
void postOrder(TreeNode root) {
if (root == null) return;
// 访问优先级:左子树 -> 右子树 -> 根结点
postOrder(root.left);
postOrder(root.right);
list.add(root.val);
}
``` ```
=== "C++" === "C++"

View file

@ -68,3 +68,9 @@ body {
--md-text-font-family: -apple-system,BlinkMacSystemFont,var(--md-text-font,_),Helvetica,Arial,sans-serif; --md-text-font-family: -apple-system,BlinkMacSystemFont,var(--md-text-font,_),Helvetica,Arial,sans-serif;
--md-code-font-family: var(--md-code-font,_),SFMono-Regular,Consolas,Menlo,-apple-system,BlinkMacSystemFont,var(--md-text-font,_),monospace; --md-code-font-family: var(--md-code-font,_),SFMono-Regular,Consolas,Menlo,-apple-system,BlinkMacSystemFont,var(--md-text-font,_),monospace;
} }
/* max height of code block */
/* https://github.com/squidfunk/mkdocs-material/issues/3444 */
.md-typeset pre > code {
max-height: 30rem;
}

View file

@ -11,6 +11,8 @@ import re
import glob import glob
import shutil import shutil
from docs.utils.extract_code_python import ExtractCodeBlocksPython from docs.utils.extract_code_python import ExtractCodeBlocksPython
from docs.utils.extract_code_java import ExtractCodeBlocksJava
def build_markdown(md_path): def build_markdown(md_path):
with open(md_path, "r") as f: with open(md_path, "r") as f:
@ -19,6 +21,7 @@ def build_markdown(md_path):
code_blocks_dict = {} code_blocks_dict = {}
file_pattern = re.compile(r'\s*```(\w+)\s+title="(.+)"') file_pattern = re.compile(r'\s*```(\w+)\s+title="(.+)"')
src_pattern = re.compile(r'\s*\[class\]\{(.*?)\}-\[func\]\{(.*?)\}') src_pattern = re.compile(r'\s*\[class\]\{(.*?)\}-\[func\]\{(.*?)\}')
for i in range(len(lines)): for i in range(len(lines)):
# Find the line target to the source codes # Find the line target to the source codes
src_match = src_pattern.match(lines[i]) src_match = src_pattern.match(lines[i])
@ -28,48 +31,43 @@ def build_markdown(md_path):
file_match = file_pattern.match(lines[j]) file_match = file_pattern.match(lines[j])
if file_match is not None: if file_match is not None:
break break
# Get the coresponding language code extractor
# Get code blocks
lang = file_match[1] lang = file_match[1]
file_name = file_match[2] file_name = file_match[2]
extractor = extractor_dict[lang]
# Get code blocks
if file_name not in code_blocks_dict: if file_name not in code_blocks_dict:
code_blocks_dict[file_name] = ExtractCodeBlocksPython( code_blocks_dict[file_name] = extractor.extract(
file_path=osp.dirname(md_path).replace("docs/", f"codes/{lang}/") + f"/{file_name}") file_path=osp.dirname(md_path).replace("docs/", f"codes/{lang}/") + f"/{file_name}")
header_line = i header_line = i
class_label = src_match[1] class_label = src_match[1]
func_label = src_match[2] func_label = src_match[2]
code_blocks = code_blocks_dict[file_name] code_blocks = code_blocks_dict[file_name]
src_info = {
"line_number": i,
"class_label": src_match[1],
"func_label": src_match[2],
"code_blocks": code_blocks_dict[file_name]
}
# Add the class to the doc # Add the class to the doc
if not func_label and class_label: if not func_label and class_label:
if class_label in code_blocks.classes: if class_label in code_blocks["classes"]:
lines.pop(header_line) lines.pop(header_line)
class_block = code_blocks.classes[class_label]["block"] class_block = code_blocks["classes"][class_label]["block"]
for code_line in class_block[::-1]: for code_line in class_block[::-1]:
ind = " " * 4 if code_line != "\n" else "" ind = " " * 4 if code_line != "\n" else ""
lines.insert(header_line, ind + code_line) lines.insert(header_line, ind + code_line)
# Add the function to the doc # Add the function to the doc
elif func_label and not class_label: elif func_label and not class_label:
if func_label in code_blocks.functions: if func_label in code_blocks["funcs"]:
lines.pop(header_line) lines.pop(header_line)
func_block = code_blocks.functions[func_label] func_block = code_blocks["funcs"][func_label]
for code_line in func_block["block"][::-1]: for code_line in func_block["block"][::-1]:
ind = " " * 4 if code_line != "\n" else "" ind = " " * 4 if code_line != "\n" else ""
lines.insert(header_line, ind + code_line) lines.insert(header_line, ind + code_line)
# Add the class method to the doc # Add the class method to the doc
elif func_label and class_label: elif func_label and class_label:
if class_label in code_blocks.classes: if class_label in code_blocks["classes"]:
class_dict = code_blocks.classes[class_label] class_dict = code_blocks["classes"][class_label]
if func_label in class_dict["functions"]: if func_label in class_dict["funcs"]:
lines.pop(header_line) lines.pop(header_line)
func_block = class_dict["functions"][func_label] func_block = class_dict["funcs"][func_label]
for code_line in func_block["block"][::-1]: for code_line in func_block["block"][::-1]:
lines.insert(header_line, code_line) lines.insert(header_line, code_line)
@ -78,10 +76,17 @@ def build_markdown(md_path):
print(f"Built {md_path}") print(f"Built {md_path}")
extractor_dict = {
"java": ExtractCodeBlocksJava(),
"python": ExtractCodeBlocksPython(),
}
if __name__ == "__main__": if __name__ == "__main__":
# Copy files to the build dir # Copy files to the build dir
shutil.copytree("docs", "build", dirs_exist_ok=True) shutil.copytree("docs", "build", dirs_exist_ok=True)
shutil.rmtree("build/utils") shutil.rmtree("build/utils")
# Build docs # Build docs
for md_path in glob.glob("docs/chapter_*/*.md"): for md_path in glob.glob("docs/chapter_*/*.md"):
build_markdown(md_path) build_markdown(md_path)

View file

@ -0,0 +1,145 @@
"""
File: extract_code_python.py
Created Time: 2023-02-07
Author: Krahets (krahets@163.com)
"""
import re
import os
import os.path as osp
class ExtractCodeBlocksJava:
def __init__(self) -> None:
self.langs = ["java"]
# Pattern to match function names and class names
self.func_pattern = r'(\s+)(public|private|)\s*(static|)\s*(\S+)\s+(\w+)(\(.*\))\s+\{'
self.class_pattern = r'(public|)\s*class\s+(\w+)\s*\{'
# Pattern to match the start and end of a block
self.block_end_pattern = '^\s{ind}\}'
self.block_start_pattern = '^\s{ind}\/\*.+\*\/'
def extract(self, file_path):
self.file_path = file_path
with open(file_path) as f:
self.lines = f.readlines()
self.content = "".join(self.lines)
# Detect and extract all the classes along with its fucntions
classes = self.extract_class_blocks()
# Remove 'static'
self.post_process(classes)
return {
"classes": classes
}
def post_process(self, classes):
for clas in classes.values():
funcs = clas["funcs"]
for func in funcs.values():
for i, line in enumerate(func["block"]):
if "static " in line:
func["block"][i] = line.replace("static ", "")
break
def search_block(self, header_line, indentation):
"""
Search class/function block given the header_line and indentation
"""
start_line, end_line = 0, len(self.lines)
block_end_pattern = re.compile(self.block_end_pattern.replace("ind", str(indentation)))
block_start_pattern = re.compile(self.block_start_pattern.replace("ind", str(indentation)))
# Search the code
for i in range(header_line + 1, len(self.lines)):
if re.match(block_end_pattern, self.lines[i]) is not None:
end_line = i
break
# Search the header comment
for i in range(header_line - 1, -1, -1):
if re.search(block_start_pattern, self.lines[i]) is not None:
start_line = i
break
code_block = self.lines[start_line:end_line + 1]
# Remove empty lines at bottom
for i in range(len(code_block) - 1, -1, -1):
if re.search("^\s*\n", code_block[i]) is None:
break
end_line -= 1
return start_line, end_line, self.lines[start_line:end_line + 1]
def extract_function_blocks(self, indentation=4, start_line=-1, end_line=-1):
"""
Extract all the functions with given indentation
"""
funcs = {}
if start_line == -1:
start_line = 0
if end_line == -1:
end_line = len(self.lines) - 1
func_pattern = re.compile(self.func_pattern)
for line_num in range(start_line, end_line + 1):
# Search the function header
func_match = func_pattern.match(self.lines[line_num])
if func_match is None: continue
# The function should match the input indentation
if len(func_match.group(1)) != indentation: continue
header_line = line_num
# Search the block from the header line
start_line, end_line, func_block = self.search_block(header_line, indentation)
# Construct the funcs dict
func_label = func_match.group(5)
funcs[func_label] = {
"indentation": indentation,
"line_number": {
"start": start_line,
"end": end_line,
"header": header_line,
},
"block": func_block,
}
return funcs
def extract_class_blocks(self):
"""
Extract all the classes with given indentation
"""
classes = {}
class_pattern = re.compile(self.class_pattern)
for line_num, line in enumerate(self.lines):
# Search the class header
class_match = class_pattern.match(line)
if class_match is None: continue
header_line = line_num
# Search the block from the header line
start_line, end_line, class_block = self.search_block(header_line, 0)
# Construct the classes dict
class_label = class_match.group(2)
classes[class_label] = {
"indentation": 0,
"line_number": {
"start": start_line,
"end": end_line,
"header": header_line,
},
"block": class_block,
"funcs": self.extract_function_blocks(
indentation=4, start_line=start_line, end_line=end_line)
}
return classes
# ext = ExtractCodeBlocksJava()
# ext.extract("codes/java/chapter_array_and_linkedlist/array.java")
# ext.extract("codes/java/chapter_array_and_linkedlist/my_list.java")

View file

@ -1,44 +1,55 @@
""" """
File: extract_code_python.py File: extract_code_python.py
Created Time: 2023-02-06 Created Time: 2023-02-07
Author: Krahets (krahets@163.com) Author: Krahets (krahets@163.com)
""" """
import re import re
import os import os
import os.path as osp import os.path as osp
import glob
class ExtractCodeBlocksPython: class ExtractCodeBlocksPython:
def __init__(self, file_path) -> None: def __init__(self) -> None:
self.langs = ["python"]
# Pattern to match function names and class names
self.func_pattern = r'(\s*)def\s+(\w+)\s*\('
self.class_pattern = r'class\s+(\w+)'
# Pattern to match the start and end of a block
self.block_end_pattern = '^\s{0,ind}\S+.*\n'
self.block_start_pattern = '^\s{ind}""".+'
def extract(self, file_path):
self.file_path = file_path self.file_path = file_path
with open(file_path) as f: with open(file_path) as f:
self.lines = f.readlines() self.lines = f.readlines()
self.content = "".join(self.lines) self.content = "".join(self.lines)
# Regular expression pattern to match function names and class names
self.func_pattern = re.compile(r'(\s*)def\s+(\w+)\s*\(')
self.class_pattern = re.compile(r'class\s+(\w+)')
# Detect and extract all the classes and fucntions # Detect and extract all the classes and fucntions
self.classes = self.extract_class_blocks() classes = self.extract_class_blocks()
self.functions = self.extract_function_blocks() funcs = self.extract_function_blocks()
return {
"classes": classes,
"funcs": funcs,
}
def search_block(self, header_line, indentation): def search_block(self, header_line, indentation):
""" """
Search class/function block given the header_line and indentation Search class/function block given the header_line and indentation
""" """
start_line, end_line = 0, len(self.lines) start_line, end_line = 0, len(self.lines)
block_end_pattern = re.compile(self.block_end_pattern.replace("ind", str(indentation)))
block_start_pattern = re.compile(self.block_start_pattern.replace("ind", str(indentation)))
# Search the code # Search the code
for i in range(header_line + 1, len(self.lines)): for i in range(header_line + 1, len(self.lines)):
if re.search("^\s*\n|^\s{ind}\s+.+\n".replace("ind", str(indentation)), if re.match(block_end_pattern, self.lines[i]) is not None:
self.lines[i]) is None:
end_line = i end_line = i
break break
# Search the header comment # Search the header comment
for i in range(header_line - 1, -1, -1): for i in range(header_line - 1, -1, -1):
if re.search('^\s{ind}""".+'.replace("ind", str(indentation)), if re.search(block_start_pattern, self.lines[i]) is not None:
self.lines[i]) is not None:
start_line = i start_line = i
break break
func_block = self.lines[start_line:end_line] func_block = self.lines[start_line:end_line]
@ -55,16 +66,18 @@ class ExtractCodeBlocksPython:
""" """
Extract all the functions with given indentation Extract all the functions with given indentation
""" """
functions = {} funcs = {}
if start_line == -1: if start_line == -1:
start_line = 0 start_line = 0
if end_line == -1: if end_line == -1:
end_line = len(self.lines) - 1 end_line = len(self.lines) - 1
func_pattern = re.compile(self.func_pattern)
for line_num in range(start_line, end_line + 1): for line_num in range(start_line, end_line + 1):
# Search the function header # Search the function header
func_match = self.func_pattern.match(self.lines[line_num]) func_match = func_pattern.match(self.lines[line_num])
if func_match is None: continue if func_match is None: continue
# The function should match the input indentation # The function should match the input indentation
if len(func_match.group(1)) != indentation: continue if len(func_match.group(1)) != indentation: continue
@ -72,9 +85,9 @@ class ExtractCodeBlocksPython:
# Search the block from the header line # Search the block from the header line
start_line, end_line, func_block = self.search_block(header_line, indentation) start_line, end_line, func_block = self.search_block(header_line, indentation)
# Construct the functions dict # Construct the funcs dict
func_label = func_match.group(2) func_label = func_match.group(2)
functions[func_label] = { funcs[func_label] = {
"indentation": indentation, "indentation": indentation,
"line_number": { "line_number": {
"start": start_line, "start": start_line,
@ -84,7 +97,7 @@ class ExtractCodeBlocksPython:
"block": func_block, "block": func_block,
} }
return functions return funcs
def extract_class_blocks(self): def extract_class_blocks(self):
""" """
@ -92,9 +105,11 @@ class ExtractCodeBlocksPython:
""" """
classes = {} classes = {}
class_pattern = re.compile(self.class_pattern)
for line_num, line in enumerate(self.lines): for line_num, line in enumerate(self.lines):
# Search the class header # Search the class header
class_match = self.class_pattern.match(line) class_match = class_pattern.match(line)
if class_match is None: continue if class_match is None: continue
header_line = line_num header_line = line_num
@ -110,7 +125,7 @@ class ExtractCodeBlocksPython:
"header": header_line, "header": header_line,
}, },
"block": class_block, "block": class_block,
"functions": self.extract_function_blocks( "funcs": self.extract_function_blocks(
indentation=4, start_line=start_line, end_line=end_line) indentation=4, start_line=start_line, end_line=end_line)
} }