信息论编码作业（香农、菲诺、赫夫曼编码）

信息论编码作业（香农、费诺、赫夫曼）

用c++实现

**写的很垃圾，代码比较冗长，仅供参考
**有一些思路是借鉴网上大佬的思路，只是作了修改
**因为交作业了就直接把代码全部发出来

香农编码

#include <iostream>
#include <math.h>
#define MAX 100
using namespace std;

void Output(double a[],int n);
//定义香农编码结构体
struct ShannonCode{
    double x;  //字符概率
    double y;  //字符概率累加和
    int k;     //码长
    int code;  //码字
}x[MAX];

//输入函数
void Input(){
    int i,m;
    double sum=0;
    double a[MAX];
    cout<<"请输入信源个数：";
    cin>>m;
    cout<<"以此输入信源概率："<<endl;
    for(i=0;i<m;i++){
        cout<<"P(x"<<i+1<<"):";
        cin>>a[i];
        sum+=a[i];
    }
    if(sum!=1){
        cout<<"概率和不为1，请重新输入！"<<endl;
        a[m]=NULL;
        Input();    //递归调用输入函数
    }
    else Output(a,m);
}

//定义输出函数，包括概率排布，累加概率，码长，码字
void Output(double a[],int n){
    int i,j,max;
    double t,p=0,L=0;
    for(i=0;i<n;i++){
         x[i].x=a[i];
    }
    //采用冒泡排序将概率从大到小排列
    for(i=0;i<n;i++){
        max=i;
        for(j=i+1;j<n;j++){
            if(x[i].x<=x[j].x)
                max=j;
            t=x[i].x;
            x[i].x=x[max].x;
            x[max].x=t;
        }
    }
    //计算码长
    for(i=0;i<n;i++){
        L=-log(x[i].x)/log(2);
        if(L-(int)L==0)
            x[i].k=(int)L;
        else
            x[i].k=(int)L+1;
    }
    cout<<endl;
    cout<<"信源符号xi"<<"\t概率P(xi)"<<"\t累加概率和"<<"\t码长"<<"\t码字"<<endl;
    for(i=0;i<n;i++){
        //计算累加概率和
        if(i!=0){
            p+=x[i-1].x;
            x[i].y=p;
        }
        else
            x[i].y=0;
        cout<<" "<<"S"<<i+1<<"\t\t"<<x[i].x<<"\t\t"<<x[i].y<<"\t\t"<<x[i].k<<"\t";

        //计算二元码长
        float t=x[i].y;
        for(j=0;j<x[i].k;j++){
            t=t*2;
            if(t>=1){
                cout<<"1";
                t-=1;
            }
            else{
                cout<<"0";
            }
        }
        cout<<endl;
    }
    //计算信源熵,平均码长
    double H=0,h,K=0;
    for(i=0;i<n;i++){
        h=-x[i].x*log(x[i].x)/log(2);
        H+=h;
        K+=x[i].x*x[i].k;
    }
    //计算编码效率
    double Y;
    Y=H/K;
    cout<<endl;
    cout<<"信源熵H(X):"<<H<<"\t平均码长为："<<K<<"\t编码效率为："<<Y<<endl;
}
int main()
{
    Input();
    return 0;
}

**结果：
**
**费诺编码

#include <iostream>
#include <stdlib.h>
#include <math.h>
#define MAX 100
using namespace std;

void Output(double a[],int n);

int index=0;
//定义费诺编码的结构
struct fano{
    double x;  //字符概率
    int k;     //码长
    int node[100];  //码字
}x[MAX];

//输入函数
void Input(){
    int i,m;
    double sum=0;
    double a[MAX];
    cout<<"请输入信源个数：";
    cin>>m;
    cout<<"以此输入信源概率："<<endl;
    for(i=0;i<m;i++){
        cout<<"P(x"<<i+1<<"):";
        cin>>a[i];
        sum+=a[i];
    }
    if(sum!=1){
        cout<<"概率和不为1，请重新输入！"<<endl;
        a[m]=NULL;
        Input();    //递归调用输入函数
    }
    else Output(a,m);
}

//确定分界点
int function1(int a,int b){
    int d=a;   //分界点的编号
    int i,j;
    double sum1=0,sum2=0,num[MAX];
    for(i=a;i<b;i++){
        for(j=a;j<i+1;j++)
            sum1+=x[j].x;
        for(j=i+1;j<b;j++)
            sum2+=x[j].x;
        num[i]=fabs(sum1-sum2);
        sum1=0;
        sum2=0;
    }
    for(i=a;i<b;i++){
        if(num[d]>num[i])
            d=i;
    }
    return d+1;
}
 //将概率分组并求出码长和码字
 void function2(int a,int num1,int b,int t){
     for(int i=a;i<num1;i++){
        x[i].node[t]=0;
        x[i].k+=1;
     }
     for(int i=num1;i<b;i++){
        x[i].node[t]=1;
        x[i].k+=1;
     }
 }
 //菲诺编码实现函数
 void function3(int c,int n){
    int num1=function1(c,n);
    function2(c,num1,n,index);
    index++;
    if(num1-c>1)
        function3(c,num1);
    if(n-num1>1)
        function3(num1,n);
    index--;
 }
//定义输出函数，包括概率排布，码长，码字
void Output(double a[],int n){
    int i,j,max;
    double t;
    for(i=0;i<n;i++){
         x[i].x=a[i];
    }
    //采用冒泡排序将概率从大到小排列
    for(i=0;i<n;i++){
        max=i;
        for(j=i+1;j<n;j++){
            if(x[i].x<=x[j].x)
                max=j;
            t=x[i].x;
            x[i].x=x[max].x;
            x[max].x=t;
        }
    }
    function3(0,n);
    cout<<endl;
    cout<<"信源符号xi"<<"\t信源概率"<<"\t码长"<<"\t码字"<<endl;
    for(i=0;i<n;i++){
         cout<<" "<<"S"<<i+1<<"\t\t"<<x[i].x<<"\t\t"<<x[i].k<<"\t";
         for(j=0;j<x[i].k;j++)
            cout<<x[i].node[j];
         cout<<endl;
    }
     //计算信源熵,平均码长
    double H=0,h,K=0;
    for(i=0;i<n;i++){
        h=-x[i].x*log(x[i].x)/log(2);
        H+=h;
        K+=x[i].x*x[i].k;
    }
    //计算编码效率
    double Y;
    Y=H/K;
    cout<<endl;
    cout<<"信源熵H(X):"<<H<<"\t平均码长为："<<K<<"\t编码效率为："<<Y<<endl;
}
int main()
{
    Input();
    return 0;
}

**结果：

**
赫夫曼编码