给定两个矩阵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 <iostream>
using namespace std;
#define MAX 100
typedef struct node{
int row,col;
int value;
}Node;
int main()
{
int Row,Col;
int An1,Bn1;
int i,N;
Node Matrix_A[MAX];
Node Matrix_B[MAX];
Node Matrix_C[MAX];
int index_A=0,index_B=0,index_C=0;
int value;
cin>>Row>>Col; //输入部分
N = Row*Col;
cin>>An1;
for(i=0;i<An1;i++)
{
cin>>Matrix_A[i].row>>Matrix_A[i].col>>Matrix_A[i].value;
}
cin>>Bn1;
for(i=0;i<Bn1;i++)
{
cin>>Matrix_B[i].row>>Matrix_B[i].col>>Matrix_B[i].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)
{
Matrix_C[index_C]=Matrix_B[index_B];
index_C++;
index_B++;
}else{
if(Matrix_A[index_A].col<Matrix_B[index_B].col)
{
Matrix_C[index_C]=Matrix_A[index_A];
index_C++;
index_A++;
}else if(Matrix_A[index_A].col>Matrix_B[index_B].col)
{
Matrix_C[index_C]=Matrix_B[index_B];
index_C++;
index_B++;
}else
{
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++;
}
}
} //while
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(i=0;i<index_C;i++)
{
cout<<Matrix_C[i].row<<" "<<Matrix_C[i].col<<" "<<Matrix_C[i].value<<endl;
}
return 0;
}
总结:
代码实现方法比较笨,核心代码部分是对各种情况的一一列举(例如,A的行数小于B,大于B,等于B的情况。在等于B的情况下,再判断A、B的列的情况。)
带所有判断结束后,在将A、B种没有尽数赋值到C的一一赋值。总结来说代码较笨,却也较容易理解。