题目:
实现三元组表示的两个稀疏矩阵的加法。
相关定义如下:
ElemType elem[MAXSIZE];
int last;
};
typedef struct _seqlist SeqList;
在三元组中,i 和 j 从 1 开始计数,与数学中矩阵元素的编号一致
矩阵加法函数的原型为:
bool add_matrix(const TSMatrix *pM, const TSMatrix *pN, TSMatrix *pQ);
pM, pN, pQ 分别指向三个矩阵,当 pM 和 pN 两个矩阵不可加时,函数返回 false,否则函数返回 true,且 pQ 指向两个矩阵的和。
代码:
#include <stdio.h>
#include <stdlib.h>
#include "tsmatrix.h"
bool add_matrix(const TSMatrix* pM, const TSMatrix* pN, TSMatrix* pQ)
{
if (pM->m != pN->m || pM->n != pN->n) {
return false;
} //判断两矩阵是否同型,同型才可做加法,否则直接返回false
int i, j, tot = 0;
int temp[MAXSIZE + 1][MAXSIZE + 1] = { 0 }; //新开二维数组temp暂时存储结果,后续导入Q中
for (i = 0; i < pM->len; i++) {
temp[pM->data[i].i][pM->data[i].j] += pM->data[i].e;
}
for (i = 0; i < pN->len; i++) {
temp[pN->data[i].i][pN->data[i].j] += pN->data[i].e;
} //将两矩阵和存储到Q中
for (i = 1; i <= pM->m; i++) {
for (j = 1; j <= pM->n; j++) {
if (temp[i][j] != 0) { //稀疏矩阵只存储值不为0的结果
pQ->data[tot++].e = temp[i][j]; //为Q赋element值
pQ->data[tot - 1].i = i;
pQ->data[tot - 1].j = j; //为Q赋行数列数
}
}
}
pQ->len = tot; //赋长度
return true;
}