MATLAB构成霍夫曼树代码

MATLAB中构建霍夫曼树的代码如下：

function huffman_tree = build_huffman_tree(symbols, probabilities)
    % 创建叶子节点
    num_symbols = length(symbols);
    leaf_nodes = cell(num_symbols, 1);
    for i = 1:num_symbols
        leaf_nodes{i} = struct('symbol', symbols(i), 'probability', probabilities(i));
    end
    
    % 构建霍夫曼树
    while length(leaf_nodes) > 1
        % 按概率从小到大排序
        [~, sorted_indices] = sort(cellfun(@(x) x.probability, leaf_nodes));
        sorted_leaf_nodes = leaf_nodes(sorted_indices);
        
        % 取出概率最小的两个节点
        node1 = sorted_leaf_nodes{1};
        node2 = sorted_leaf_nodes{2};
        
        % 创建新节点，概率为两个节点概率之和
        new_node = struct('left', node1, 'right', node2, 'probability', node1.probability + node2.probability);
        
        % 从叶子节点列表中移除已处理的节点，添加新节点
        leaf_nodes = sorted_leaf_nodes(3:end);
        leaf_nodes{end+1} = new_node;
    end
    
    % 返回霍夫曼树的根节点
    huffman_tree = leaf_nodes{1};
end

这段代码实现了构建霍夫曼树的功能。它接受两个输入参数：`symbols`表示符号集合，`probabilities`表示每个符号对应的概率。代码首先创建叶子节点，每个叶子节点包含一个符号和对应的概率。然后，通过循环构建霍夫曼树，每次从叶子节点列表中选择概率最小的两个节点，创建一个新节点，并将其添加到列表中。最终，返回霍夫曼树的根节点。