bool add_matrix(const TSMatrix* pM, const TSMatrix* pN, TSMatrix* pQ)
{
if (pM->n != pN->n || pM->m != pN->m) {
return false;
}
int i = 0, j = 0, q = 0;
for (i = 0, j = 0; i < pM->len && j < pN->len;) {
if ((j < pN->len && i < pM->len) && pM->data[i].i == pN->data[j].i && pM->data[i].j == pN->data[j].j) {
if (pM->data[i].e + pN->data[j].e == 0) {
i++;
j++;
continue;
} else
pQ->data[q].i = pM->data[i].i;
pQ->data[q].j = pN->data[j].j;
pQ->data[q].e = pM->data[i].e + pN->data[j].e;
i++;
j++;
q++;
} else if ((j < pN->len && i < pM->len) && (pM->data[i].i - 1) * (pM->n) + pM->data[i].j < (pN->data[j].i - 1) * (pN->n) + pN->data[j].j) {
pQ->data[q].i = pM->data[i].i;
pQ->data[q].j = pM->data[i].j;
pQ->data[q].e = pM->data[i].e;
i++;
q++;
} else if ((j < pN->len && i < pM->len) && (pM->data[i].i - 1) * (pM->n) + pM->data[i].j > (pN->data[j].i - 1) * (pN->n) + pN->data[j].j) {
pQ->data[q].i = pN->data[j].i;
pQ->data[q].j = pN->data[j].j;
pQ->data[q].e = pN->data[j].e;
j++;
q++;
}
}
while (i < pM->len) {
pQ->data[q].i = pM->data[i].i;
pQ->data[q].j = pM->data[i].j;
pQ->data[q].e = pM->data[i].e;
i++;
q++;
}
while (j < pN->len) {
pQ->data[q].i = pN->data[j].i;
pQ->data[q].j = pN->data[j].j;
pQ->data[q].e = pN->data[j].e;
j++;
q++;
}
pQ->m = pN->m;
pQ->n = pN->n;
pQ->len = q;
return true;
}
05-24