迭代算法求信道容量.docVIP

C语言方法实现，源程序如下： #include stdio.h #include math.h #includestdlib.h #define N 4 #define M 4 /*转移矩阵行数为N，列数为M，自行调整*/ void Init(double *Pa); /* 输入概率分布初始化函数*/ void Input(double *Pa,double *a); /*迭代输入概率分布重新调整函数*/ void Output(double *Pa,double P[N][M],double *Pb); /*输出概率分布函数*/ void Infor(double *Pb,double P[N][M],double *a); /*计算a(i)的函数*/ double capacity1(double *Pa,double *a); /*计算C(n+1,n)*/ double capacity2(double *a); /*计算C(n+1,n)的函数*/ void main() { double c1,c2,s,temp; double Pa[N],a[N],Pb[M]; double P[N][M]; int i,j,count=0; /*count记录迭代次数*/ for(i=0;iN;i++) for(j=0;jM;j++) { printf(please input P[%d][%d]:,i+1,j+1); scanf(%lf,P[i][j]); } /*输入转移概率矩阵P*/ for(i=0;iN;i++) { temp=0.0; for(j=0;jM;j++) temp+=P[i][j]; if(temp!=1) { printf(Error!\n);exit(0);} } /*检查输入的转移概率矩阵是否正确：各行之和为1，则继续进行；否则退出程序*/ printf(请输入精度：\n); scanf(%lf,s); Init(Pa); while(1) { count++; Output(Pa,P,Pb); Infor(Pb,P,a); c1=capacity1(Pa,a); c2=capacity2(a); if(fabs(c1-c2)s) break; else Input(Pa,a); } printf(迭代次数=%d次\n,count); printf(信道容量C=%fbit/符号\n,(1.0/log(2))*c1); printf(输入概率分布为:\n); for(i=0;iM;i++) { printf(%10f,Pa[i]); if(i==N-1) printf(\n); } } void Init(double *Pa) { double s=N;int i; for(i=0;iN;i++) Pa[i]=1.0/s; } void Input(double *Pa,double *a) { int i; double temp=0.0; for(i=0;iN;i++) temp+=Pa[i]*a[i]; for(i=0;iN;i++) Pa[i]=(Pa[i]*a[i])/temp; } void Output(double *Pa,double P[N][M],double *Pb) { int i,j;double temp=0.0; for(j=0;jM;j++) { for(i=0;iN;i++) temp+=Pa[i]*P[i][j]; Pb[j]=temp; temp=0.0; } } void Infor(double *Pb,double P[N][M],double *a) { int i,j;double temp=0.0; for(i=0;iN;i++) { for(j=0;jM;j++) if(P[i][j]==0) temp+=0; else temp+=P[i][j]*log(P[i][j]/Pb[j]); a[i