3.0 稀疏矩阵加法

给定两个矩阵A和B，求其和矩阵C=A+B。

输入格式:

第一行包含两个数Row和Col，分别表示矩阵的行数和列数，A和B的维度是一致的。

第二行只有一个数N1​，表示接下来要输入的A中的非零元素的个数。

接下来是N1​行，每一行都是i j A[i,j] 这样的形式，表示的A中第i行第j列的元素A[i,j]，为了与大多数编程语言保持一致，它们都是从零开始的，也就是说下标的有效范围是[0,Row−1]×[0,Col−1]。

在N1​行之后，是一个数N2​，表示矩阵B中非零元素的数量，此后N2​行描述B中的非零元素，它们与此前描述A中非零元素的形式一致。

矩阵元素的输入均遵循行主序。这里的所有的输入均可用int类型正确表示，可以假设输入均是合法的。

输出格式:

第一行输出和矩阵C=A+B中的绝对值大于0.1的元素个数N3​，此后是N3​行，按照行主序输出其中的非零元素，依次是行、列的下标和对应的元素。

输入样例:

2 2
1
1 1 1
1
0 0 1

输出样例:

2
0 0 1
1 1 1

代码长度限制

16 KB

时间限制

400 ms

内存限制

10 MB

#include <stdio.h>
#include <string.h>
#include <math.h>
//稀疏矩阵压缩存储 注意按行主序输入输出

typedef struct matrix
{
    int i, j, x;
} matrix;

int main()
{
    int row, col;
    scanf ("%d %d", &row, &col);
    
    int n1, n2;

    scanf("%d", &n1);
    matrix a[n1];
    for (int k = 0; k < n1; k++)
    {
        int i, j;
        scanf ("%d %d %d", &a[k].i, &a[k].j, &a[k].x);
    }

    scanf("%d", &n2);
    matrix b[n2];
    for (int k = 0; k < n2; k++)
    {
        int i, j;
        scanf ("%d %d %d", &b[k].i, &b[k].j, &b[k].x);
    }

    matrix c[n1+n2];
    int cnt = 0, p = 0, q = 0;
    while (p < n1 || q < n2)
    {
        if (p >= n1)
        {
            c[cnt++] = b[q++];
            continue;
        }
        if (q >= n2)
        {
            c[cnt++] = a[p++];
            continue;
        }

        if (a[p].i*col + a[p].j < b[q].i*col + b[q].j)
            c[cnt++] = a[p++];
        else if (a[p].i*col + a[p].j > b[q].i*col + b[q].j)
            c[cnt++] = b[q++];
        else
        {
            if (a[p].x + b[q].x != 0)
            {
                c[cnt].i = a[p].i;
                c[cnt].j = a[p].j;
                c[cnt].x = a[p].x + b[q].x;
                cnt++;
            }
            p++, q++;
        }
    }

    printf("%d\n", cnt);
    for (int k = 0; k < cnt; k++)
        printf("%d %d %d\n", c[k].i, c[k].j, c[k].x);

    return 0;
}