1untitled.m

已知准对称信道转移矩阵和输入分布

clear
format  compact
disp('假设信道转移矩阵为:[1/3 1/3 1/6 1/6; 1/6 1/3 1/6 1/3] ,假定输入分布: [3/4 1/4]');
disp('请键入信道转移矩阵:');
Pb_a=input('Pba=');
disp('请键入输入分布:');
Pa0=input('Pa0=');
[r,s]=size(Pb_a);
[r0,s0]=size(Pa0);
while(s0~=r)
    Px0=input('阶数错误！请重新输入初始输入分布矩阵:');
    [r0,s0]=size(Px0);
end;
disp('请键入要求精度:');
e=input('精度为:');
Cn=0;
Cn1=10;
Pa=Pa0

while  abs(Cn1-Cn)>e,
     
    for  k=1:s0
        Md=0;
       for  j=1:s
            Me=0;
            for i=1:r 
               Me=Me+Pa(i)*Pb_a(i,j);
            end ;
            Md=Md+Pb_a(k,j)*log2(Pb_a(k,j)/Me);
        end;
        Mb(k)=exp(Md);    
    end ;
    
    for  k=1:s0
        Mc=0;
        for i=1:r
            Mc=Mc+Pa(i)*Mb(i);
        end;
        Pa(k)=Pa(k)*Mb(k)/Mc;
    end;
    
    M_Cn=0;
    for j=1:r 
       M_Cn=M_Cn+Pa(j)*Mb(j);
    end;
    Cn=log(M_Cn);
    Cn1=log(max(Mb));
    
end; 

C=Cn;
disp('信道容量为');
disp(C);
disp('最佳输入分布为');
Pa,