实现三元组表示的两个稀疏矩阵的加法。相关定义如下
#define MAXSIZE 100 //假设非零元个数的最大值为100
typedef struct {
int i,j; //非零元的行下标和列下标,i 和 j 从 1 开始计数,与数学中矩阵元素的编号一致
ElemType e; //非零元的值
}Triple;
typedef struct {
Triple data[MAXSIZE]; // 非零元三元组表
int m, n, len; // 矩阵的行数、列数和非零元个数
}TSMatrix;
在三元组中,i 和 j 从 1 开始计数,与数学中矩阵元素的编号一致
矩阵加法函数的原型为:
pM, pN, pQ 分别指向三个矩阵,当 pM 和 pN 两个矩阵不可加时,函数返回 false,否则函数返回 true,且 pQ 指向两个矩阵的和。
pM, pN, pQ 分别指向三个矩阵,当 pM 和 pN 两个矩阵不可加时,函数返回 false,否则函数返回 true,且 pQ 指向两个矩阵的和。
bool add_matrix(const TSMatrix* pM, const TSMatrix* pN, TSMatrix* pQ)
{
if (pM->m != pN->m || pM->n != pN->n) {
return false;
}
int i, j, tot = 0;
int temp[MAXSIZE + 1][MAXSIZE + 1] = { 0 };
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;
}
for (i = 1; i <= pM->m; i++) {
for (j = 1; j <= pM->n; j++) {
if (temp[i][j] != 0) {
pQ->data[tot++].e = temp[i][j];
pQ->data[tot - 1].i = i;
pQ->data[tot - 1].j = j;
}
}
}
pQ->len = tot;
return true;
}
如果有更好的方法或者以上程序有任何错误,欢迎讨论。