稀疏矩阵相加减

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

 

给定两个矩阵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

 

结尾无空行

#include <bits/stdc++.h>
using namespace std;
struct node
{
    int x,y,z;
};
static bool cmp(node & a,node & b)
{
    if(a.x<b.x)return true;
    if(a.x==b.x&&a.y<b.y)return true;
    return false;
}
int main()
{
    int n,m;
    cin>>n>>m;
    vector<vector<int>> a,b;
    int k1;
    cin>>k1;
    for(int i=0;i<k1;i++){
        int x,y,z;
        cin>>x>>y>>z;
        vector<int> t;
        t.push_back(x);
        t.push_back(y);
        t.push_back(z);
        a.push_back(t);
    }
    int k2;
    cin>>k2;
    for(int i=0;i<k2;i++){
        int x,y,z;
        cin>>x>>y>>z;
        vector<int> t;
        t.push_back(x);
        t.push_back(y);
        t.push_back(z);
        
        b.push_back(t);
    }
    vector<node> ans;
    for(int i=0;i<k1;i++)
    {
       for(int j=0;j<k2;j++)
       {
           if(a[i][0]==b[j][0]&&a[i][1]==b[j][1])a[i][2]+=b[j][2];
       }
        node t;
        t.x=a[i][0];
        t.y=a[i][1];
        t.z=a[i][2];
        if(abs(t.z)>0.1)
        ans.push_back(t) ;
    }
    
    for(int j=0;j<k2;j++)
    {
        int flag=0;
        for(int i=0;i<k1;i++)
        {
             if(a[i][0]==b[j][0]&&a[i][1]==b[j][1])
             {
                 flag=1;
                 break;
             }
        }
        node t;
        t.x=b[j][0];
        t.y=b[j][1];
        t.z=b[j][2];
        if(flag==0)ans.push_back(t);
    }
    sort(ans.begin(),ans.end(),cmp);
    cout<<ans.size()<<endl;
    for(int i=0;i<ans.size();i++)
    {
        cout<<ans[i].x<<' '<<ans[i].y<<' '<<ans[i].z;
        if(i<ans.size()-1)cout<<endl;
    }
    return 0;
}