mirror of
https://github.com/krahets/hello-algo.git
synced 2025-11-02 12:58:42 +08:00
feat(csharp): add csharp code for array binary tree (#632)
This commit is contained in:
@ -13,50 +13,59 @@ public class TreeNode {
|
||||
public TreeNode? left; // 左子节点引用
|
||||
public TreeNode? right; // 右子节点引用
|
||||
|
||||
/* 构造方法 */
|
||||
public TreeNode(int x) {
|
||||
val = x;
|
||||
}
|
||||
|
||||
/* Generate a binary tree given an array */
|
||||
public static TreeNode? ListToTree(List<int?> arr) {
|
||||
if (arr.Count == 0 || arr[0] == null)
|
||||
return null;
|
||||
// 序列化编码规则请参考:
|
||||
// https://www.hello-algo.com/chapter_tree/array_representation_of_tree/
|
||||
// 二叉树的数组表示:
|
||||
// [1, 2, 3, 4, None, 6, 7, 8, 9, None, None, 12, None, None, 15]
|
||||
// 二叉树的链表表示:
|
||||
// /——— 15
|
||||
// /——— 7
|
||||
// /——— 3
|
||||
// | \——— 6
|
||||
// | \——— 12
|
||||
// ——— 1
|
||||
// \——— 2
|
||||
// | /——— 9
|
||||
// \——— 4
|
||||
// \——— 8
|
||||
|
||||
TreeNode root = new TreeNode(arr[0]!.Value);
|
||||
Queue<TreeNode> queue = new Queue<TreeNode>();
|
||||
queue.Enqueue(root);
|
||||
int i = 0;
|
||||
while (queue.Count != 0) {
|
||||
TreeNode node = queue.Dequeue();
|
||||
if (++i >= arr.Count) break;
|
||||
if (arr[i] != null) {
|
||||
node.left = new TreeNode((int)arr[i]);
|
||||
queue.Enqueue(node.left);
|
||||
}
|
||||
if (++i >= arr.Count) break;
|
||||
if (arr[i] != null) {
|
||||
node.right = new TreeNode((int)arr[i]);
|
||||
queue.Enqueue(node.right);
|
||||
}
|
||||
/* 将列表反序列化为二叉树:递归 */
|
||||
private static TreeNode? ListToTreeDFS(List<int?> arr, int i) {
|
||||
if (i < 0|| i >= arr.Count || !arr[i].HasValue) {
|
||||
return null;
|
||||
}
|
||||
TreeNode root = new TreeNode(arr[i].Value);
|
||||
root.left = ListToTreeDFS(arr, 2 * i + 1);
|
||||
root.right = ListToTreeDFS(arr, 2 * i + 2);
|
||||
return root;
|
||||
}
|
||||
|
||||
/* Serialize a binary tree to a list */
|
||||
public static List<int?> TreeToList(TreeNode root) {
|
||||
List<int?> list = new();
|
||||
if (root == null) return list;
|
||||
Queue<TreeNode?> queue = new();
|
||||
while (queue.Count != 0) {
|
||||
TreeNode? node = queue.Dequeue();
|
||||
if (node != null) {
|
||||
list.Add(node.val);
|
||||
queue.Enqueue(node.left);
|
||||
queue.Enqueue(node.right);
|
||||
} else {
|
||||
list.Add(null);
|
||||
}
|
||||
/* 将列表反序列化为二叉树 */
|
||||
public static TreeNode? ListToTree(List<int?> arr) {
|
||||
return ListToTreeDFS(arr, 0);
|
||||
}
|
||||
|
||||
/* 将二叉树序列化为列表:递归 */
|
||||
private static void TreeToListDFS(TreeNode? root, int i, List<int?> res) {
|
||||
if (root == null)
|
||||
return;
|
||||
while (i >= res.Count) {
|
||||
res.Add(null);
|
||||
}
|
||||
return list;
|
||||
res[i] = root.val;
|
||||
TreeToListDFS(root.left, 2 * i + 1, res);
|
||||
TreeToListDFS(root.right, 2 * i + 2, res);
|
||||
}
|
||||
|
||||
/* 将二叉树序列化为列表 */
|
||||
public static List<int?> treeToList(TreeNode root) {
|
||||
List<int?> res = new List<int?>();
|
||||
TreeToListDFS(root, 0, res);
|
||||
return res;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user