对称矩阵的压缩存储及基本运算（3）

/*
*Copyright (c) 2015 , 烟台大学计算机学院
*All right resvered .
*文件名称: 对称矩阵压缩存储的实现与应用.cpp
*作    者: 郑兆涵
*对称矩阵的压缩存储及基本运算(3)
*/

#include <stdio.h>
#include <malloc.h>
#define N 4
//为N阶对称矩阵初始化存储数据的一维数组B
void Init(int *&b)
{
b = (int*)malloc(sizeof(int)*(N*(N+1)/2));
}
//将A[i][j]的值e存储到b中
void Assign (int b[],int e,int i,int j)
{
int k;
k=(i>=j)?((i*(i+1))/2+j):(N*(N+1)/2);
b[k]=e;
return;
}
//返回存储在b[M]中的A[i][j]的值
int Value(int b[],int i,int j)
{
int k;
k=(i>=j)?((i*(i+1))/2+j):(N*(N+1)/2);
return b[k];
}

void Disp(int b[])
{
int i,j;
for (i=0; i<N; i++)
{
for (j=0; j<N; j++)
printf("%4d",Value(b,i,j));
printf("\n");
}
}

//销毁存储空间
void Destroy(int b[])
{
free(b);
}

int main()
{
int *b1;  //指向整型的指针，待初始化
int i, j;
int v;
Init(b1);
printf("请输入对称矩阵(只需要输入下三角部分即可)\n");
for(i=0;i<N;i++)
{
printf("输入第%d行的%d个数据元素: ", i+1, i+1);
for(j=0; j<=i; j++)
{
scanf("%d", &v);
Assign(b1, v, i, j);
}
}
Disp(b1);
Destroy(b1);
return 0;
}