ru/codes/zig/chapter_tree/binary_tree_dfs.zig - main - githubmirror/hello-algo

ru/codes/zig/chapter_tree/binary_tree_dfs.zig
70 строк3 KB
Yudong Jin
Add ru version (#1865)
27 мар 2026, 23:24
27 мар 2026, 23:247721837
// File: binary_tree_dfs.zig
// Created Time: 2023-01-15
// Author: codingonion (coderonion@gmail.com)

const std = @import("std");
const inc = @import("include");

var list = std.ArrayList(i32).init(std.heap.page_allocator);

// Предварительный обход
fn preOrder(comptime T: type, root: ?*inc.TreeNode(T)) !void {
    if (root == null) return;
    // Порядок обхода: корень -> левое поддерево -> правое поддерево
    try list.append(root.?.val);
    try preOrder(T, root.?.left);
    try preOrder(T, root.?.right);
}

// Симметричный обход
fn inOrder(comptime T: type, root: ?*inc.TreeNode(T)) !void {
    if (root == null) return;
    // Порядок обхода: левое поддерево -> корень -> правое поддерево
    try inOrder(T, root.?.left);
    try list.append(root.?.val);
    try inOrder(T, root.?.right);
}

// Обратный обход
fn postOrder(comptime T: type, root: ?*inc.TreeNode(T)) !void {
    if (root == null) return;
    // Порядок обхода: левое поддерево -> правое поддерево -> корень
    try postOrder(T, root.?.left);
    try postOrder(T, root.?.right);
    try list.append(root.?.val);
}

// Driver Code
pub fn main() !void {
    // Инициализация аллокатора памяти
    var mem_arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
    defer mem_arena.deinit();
    const mem_allocator = mem_arena.allocator();

    // Инициализировать двоичное дерево
    // Здесь используется функция, напрямую строящая двоичное дерево из массива
    var nums = [_]i32{1, 2, 3, 4, 5, 6, 7};
    var root = try inc.TreeUtil.arrToTree(i32, mem_allocator, &nums);
    std.debug.print("Инициализация двоичного дерева\n", .{});
    try inc.PrintUtil.printTree(root, null, false);

    // Предварительный обход
    list.clearRetainingCapacity();
    try preOrder(i32, root);
    std.debug.print("\nПоследовательность печати узлов при предварительном обходе = ", .{});
    inc.PrintUtil.printList(i32, list);

    // Симметричный обход
    list.clearRetainingCapacity();
    try inOrder(i32, root);
    std.debug.print("\nПоследовательность печати узлов при симметричном обходе = ", .{});
    inc.PrintUtil.printList(i32, list);

    // Обратный обход
    list.clearRetainingCapacity();
    try postOrder(i32, root);
    std.debug.print("\nПоследовательность печати узлов при обратном обходе = ", .{});
    inc.PrintUtil.printList(i32, list);

    _ = try std.io.getStdIn().reader().readByte();
}