第8周 项目3- 对称矩阵压缩存储的实现与应用(3)

/* 
* Copyright(c) 2017,烟台大学计算机学院 
* All rights reserved. 
* 文件名称:a.cpp 
* 作    者:陈军正
* 完成日期:2017 年 10 月 19 日 
* 版 本 号:v1.0 
* 
* 问题描述:写出对上、下三角矩阵、对角矩阵的压缩存储结构，以及相关的基本运算的实现 
* 输入描述:请输入对称矩阵(只需要输入下三角部分) 
* 程序输出:实现各种算法的函数的测试结果 
*/  
#include <stdio.h>  
#include <malloc.h>  
#define N 4  
#define M 10  
void Init(int *&b)  
{  
    b = (int*)malloc(sizeof(int)*(N*(N+1)/2));  
}  
  
//返回存储在b[M]中，对应二维数组A[i][j]的值  
int Value(int b[], int i, int j)  
{  
    if (i>=j)  
        return b[(i*(i+1))/2+j];  
    else  
        return b[(j*(j+1))/2+i];  
}  
  
//将e赋值给对应二维数组元素A[i][j]，要存储到B[M]中  
void Assign(int b[], int e, int i, int j)  
{  
    if (i>=j)  
        b[(i*(i+1))/2+j] = e;  
    else  
        b[(j*(j+1))/2+i] = e;  
    return;  
}  
  
//输出压缩存储在b中的对称矩阵  
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 value(int a[],int i,int j)  
{  
    if (i>=j)  
        return a[(i*(i+1))/2+j];  
    else  
        return a[(j*(j+1))/2+i];  
}  
void madd(int a[],int b[],int c[][N])  
{  
    int i,j;  
    for (i=0; i<N; i++)  
        for (j=0; j<N; j++)  
            c[i][j]=value(a,i,j)+value(b,i,j);  
}  
void mult(int a[],int b[],int c[][N])  
{  
    int i,j,k,s;  
    for (i=0; i<N; i++)  
        for (j=0; j<N; j++)  
        {  
            s=0;  
            for (k=0; k<N; k++)  
                s=s+value(a,i,k)*value(b,k,j);  
            c[i][j]=s;  
        }  
}  
void disp1(int a[])  
{  
    int i,j;  
    for (i=0; i<N; i++)  
    {  
        for (j=0; j<N; j++)  
            printf("%4d",value(a,i,j));  
        printf("\n");  
    }  
}  
void disp2(int c[][N])  
{  
    int i,j;  
    for (i=0; i<N; i++)  
    {  
        for (j=0; j<N; j++)  
            printf("%4d",c[i][j]);  
        printf("\n");  
    }  
}  
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);  
    int a[M]= {1,2,3,4,5,6,7,8,9,10};  //a表示压缩存储的对称矩阵  
    int b[M]= {1,1,1,1,1,1,1,1,1,1};  
    int c1[N][N],c2[N][N];  
    madd(a,b,c1);  
    mult(a,b,c2);  
    printf("a矩阵:\n");  
    disp1(a);  
    printf("b矩阵:\n");  
    disp1(b);  
    printf("a+b:\n");  
    disp2(c1);  
    printf("a×b:\n");  
    disp2(c2);  
    printf("\n");  
    return 0;  
}  

运行结果