C语言 对称矩阵 压缩 实现

对称矩阵压缩的简单实现 (GCC编译)。

/**
* @brief C语言 对称矩阵 压缩 实现
* @author wid
* @date 2013-11-03
*
* @note 若代码存在 bug 或程序缺陷, 请留言反馈, 谢谢!
*/

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>

#define MAT_ROW 6
#define MAT_COL 6

typedef int ElemType;

///方法声明
ElemType *Compress( ElemType pArr2D[][MAT_COL] );       ///压缩二维对称矩阵
ElemType Value( ElemType *pComArr, int m, int n );      ///从压缩后的矩阵中取值


///方法实现
/**
* @brief 压缩对称矩阵
*
* @param 待压缩的矩阵
*
* @return 返回压缩后线性表的指针
*/
ElemType *Compress( ElemType pArr2D[][MAT_COL] )
{
    ElemType *pMat = (ElemType *)malloc( sizeof(ElemType) * ((MAT_COL * (MAT_COL + 1)) / 2) );

    ///映射到线性表中
    int m = 0, n = 0, nLen = 1, nPos = 0;
    for( m = 0; m < MAT_ROW; ++m )
    {
        for( n = 0; n < nLen; ++n )
        {
            pMat[nPos] = pArr2D[m][n];
            ++nPos;
        }

        ++nLen;
    }

    return pMat;
}

/**
* @brief 从压缩后得到的线性表中取值
*
* @param pComArr 指向存放压缩矩阵的线性表
* @param m 矩阵的第一维下标
* @param n 矩阵的第二维下标
*
* @return 返回该下标指向的元素值
*
* @note 元素位置由 0 计起
*/
ElemType Value( ElemType *pComArr, int m, int n )
{
    ///使用断言
    assert( m >= 0 && m < MAT_ROW &&  n >= 0 && n < MAT_COL );

    ///将0索引转化为1索引
    ++m;
    ++n;

    ///根据对称矩阵性质从压缩一维数组中取值
    if( m >= n )
        return pComArr[ (m * (m - 1)) / 2 + n - 1 ];
    else
        return pComArr[ (n * (n - 1)) / 2 + m - 1 ];
}


///测试
int main()
{
    ///对称矩阵 arrMat
    ElemType arrMat[MAT_ROW][MAT_COL] = {
        {1, 2, 3, 4, 5, 6},
        {2, 2, 3, 4, 4, 5},
        {3, 3, 3, 4, 4, 4},
        {4, 4, 4, 3, 3, 3},
        {5, 4, 4, 3, 2, 2},
        {6, 5, 4, 3, 2, 1}
    };

    ///对矩阵 arrMat 进行压缩
    ElemType *pCom = Compress( arrMat );

    ///从压缩矩阵输出整个矩阵, 使用 0 索引
    int m = 0, n = 0;
    for( m = 0; m < MAT_ROW; ++m )
    {
        for( n = 0; n < MAT_COL; ++n )
        {
            printf( "%d ", Value(pCom, m, n) );
        }

        putchar( '
' );
    }

    return 0;
}

运行测试:

若代码存在 bug 或程序缺陷, 请留言反馈, 谢谢。