m06_huffman.m

MATLAB仿真技术与实例应用教程

function [h,l]=m06_huffman(P);
%HUFFMAN Huhhman code generator

if length(find(P<0))~=0,        %判断是否符合概率分布的条件 
    error('Not a prob.vector')
end
if abs(sum(P)-1)>10e-10,
    error('Not a prob.vector')
end
n=length(P);            %计算输入元素个数
Q=P;
m=zeros(n-1,n);
for i=1:n-1
    [Q,l]=sort(Q);      %对输入元素排序
    m(i,:)=[l(1:n-i+1),zeros(1,i-1)];
    Q=[Q(1)+Q(2),Q(3:n),1];
end
for i=1:n-1
    c(i,:)=blanks(n*n);
end
%以下计算各个元素码字
c(n-1,n)='0';
c(n-1,2*n)='1';
for i=2:n-1
    c(n-i,1:n-1)=c(n-i+1,n*(find(m(n-i+1,:)==1))-(n-2):n*(find(m(n-i+1,:)==1)));
    c(n-i,n)='0';
    c(n-i,n+1:2*n-1)=c(n-i,1:n-1);
    c(n-i,2*n)='1';
    for j=1:i-1
    c(n-i,(j+1)*n+1:(j+2)*n)=c(n-i+1,n*(find(m(n-i+1,:)==j+1)-1)+1:n*find(m(n-i+1,:)==j+1));
    end
end
for i=1:n
    h(i,1:n)=c(1,n*(find(m(1,:)==i)-1)+1:find(m(1,:)==i)*n);
    ll(i)=length(find(abs(h(i,:))~=32));
end
l=sum(P.*ll);       %计算平均码长