end

end

for i=1:r

for j=1:s

Pba(i,j,2)=(Pb(j,1)*exp(S(m)*d(i,j)))/temp(i);

end

end

D(1)=0;

for i= 1:r

for j=1:s

D(1)=D(1)+Pa(i)*Pba(i,j,1)*d(i,j); % 求D(1)

end

end

R(1)=0;

for i=1:r

for j=1:s

if (Pba(i,j, 1 ) ~=0)

R(1)=R(1)+Pa(i)*Pba(i,j,1)*log2(Pba(i,j,1)/Pb(j,1)); % 求R(1) end

end

end

n=2; %n 为内层循环次数

while(1) %内层循环，对精度的循环

%第三步

for j=1:s

Pb(j,n) = 0;

for i= 1:r

Pb(j,n)=Pb(j,n)+Pa(i)*Pba(i,j,n); % 求输出的信源概率分布 end

end

for i=1 :r

temp(i)=0;

for j=1:s

temp(i) =temp (i)+Pb (j,n) *exp(S(m) *d(i,j));

%temp为临时项，求Pba(i,j,n+1)时表达式的分母

end

end

for i=1:r

for j=1:s

if(temp(i) ~=0)

Pba(i,j,n+1)=(Pb(j, n)*exp(S(m)*d(i,j)))/temp(i); %求Pba(i, j, n+1)

end