PTA 7-3 稀疏矩阵加法 (20 分)

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

输入格式:

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

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

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

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

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

输出格式:

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

输入样例:

2 2
1
1 1 1
1
0 0 1

输出样例:

2
0 0 1
1 1 1

#include<stdio.h>
#include <iostream>

using namespace std;
const int MAX = 100;
typedef struct node{
    int row,col; //非零元素的行下标和列下标
    int value;//非零元素值
}Node;

int main()
{
    int Row,Col;
    int An1;//A的非零元素个数
    int Bn1;//B的非零元素个数
    cin >> Row >> Col;//行和列
    cin >> An1;
    int N = Row*Col;
    Node Matrix_A[MAX];
    for(int i = 0;i<An1;i++)
    {
        cin >> Matrix_A[i].row >>Matrix_A[i].col >> Matrix_A[i].value;
    }
    cin >> Bn1;
    Node Matrix_B[MAX];
    for(int i = 0;i<Bn1;i++)
    {
        cin >> Matrix_B[i].row >>Matrix_B[i].col >> Matrix_B[i].value;
    }
    Node Matrix_C[MAX];
    int index_A = 0,index_B = 0,index_C = 0;
    int value;
    while(index_A < An1&&index_B<Bn1)
    {
        if(Matrix_A[index_A].row<Matrix_B[index_B].row)
        {
            Matrix_C[index_C] = Matrix_A[index_A];
            index_C++;
            index_A++;
        }
        else if(Matrix_A[index_A].row == Matrix_B[index_B].row)
        {
            if(Matrix_A[index_A].col==Matrix_B[index_B].col)
            {
                value = Matrix_A[index_A].value+Matrix_B[index_B].value;
                if(value==0)
                {
                    index_A++;
                    index_B++;
                    continue;
                }
                Matrix_C[index_C].row = Matrix_A[index_A].row;
                Matrix_C[index_C].col = Matrix_A[index_A].col;
                Matrix_C[index_C].value = value;
                index_A++;
                index_B++;
                index_C++;
            }
            else
            {
                if(Matrix_A[index_A].col < Matrix_B[index_B].col)
                {
                    Matrix_C[index_C] = Matrix_A[index_A];
                    index_A++;
                    index_C++;
                }
                else
                {
                    Matrix_C[index_C] = Matrix_B[index_B];
                    index_B++;
                    index_C++;
                }
            }
        }
        else
        {
            Matrix_C[index_C] = Matrix_B[index_B];
            index_C++;
            index_B++;
        }
    }
    if(index_A<An1)
    {
        for(int i = index_A;i<An1;i++)
        {
            Matrix_C[index_C] = Matrix_A[i];
            index_C++;
        }
    }
    if(index_B<Bn1)
    {
        for(int i = index_B;i<Bn1;i++)
        {
            Matrix_C[index_C] = Matrix_B[i];
            index_C++;
        }
    }


    cout << index_C << endl;;
    for(int i = 0;i<index_C;i++)
    {
        printf("%d %d %d\n",Matrix_C[i].row,Matrix_C[i].col,Matrix_C[i].value);
    }

    return 0;
}

 我其实对PTA的测试方法比较迷，我开始是把Matrix数组初始化为Matrix_A[Row*Col],但是测试的时候总是出错，然后我就改成了Matrix_A[MAX]常量形式定义的数组，就完美通过...