稀疏矩阵加法(PTA)

稀疏矩阵加法

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

输入格式：

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

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

接下来是N1行，每一行都是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类型正确表示，可以假设输入均是合法的。

因为输入合法，所以第一行的Row和Col没有"意义"。

输出格式：

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

输入样例：

2 2
1
1 1 1
1
0 0 1

输出样例：

2
0 0 1
1 1 1

参考代码如下：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>

typedef struct{
	int i, j;
	int v;
}node;

int main(void){
	node A[1000], B[1000], C[1000];
	int Row, Col;
	int N1, N2, i, j, k;
	scanf("%d %d", &Row, &Col);
	//N1
	scanf("%d", &N1);
	//A[] input
	for(i = 0; i < N1; i++){
		scanf("%d %d %d", &A[i].i, &A[i].j, &A[i].v);
	}
	//N2
	scanf("%d", &N2);
	//B[] input
	for(i = 0; i < N2; i++){
		scanf("%d %d %d", &B[i].i, &B[i].j, &B[i].v);
	}
	i = 0;	//A[]
	j = 0;	//B[]
	k = 0;	//C[]

	while(i < N1 && j < N2){
		if(A[i].i < B[j].i){
			C[k].i = A[i].i;
			C[k].j = A[i].j;
			C[k].v = A[i].v;
			k++;
			i++;
		}
		else if(B[j].i < A[i].i){
			C[k].i = B[j].i;
			C[k].j = B[j].j;
			C[k].v = B[j].v;
			k++;
			j++;			
		}
		else{	
		//A[i].i == B[j].i
			if(A[i].j < B[j].j){
				C[k].i = A[i].i;
				C[k].j = A[i].j;
				C[k].v = A[i].v;
				k++;
				i++;
			}
			else if(B[j].j < A[i].j){
				C[k].i = B[j].i;
				C[k].j = B[j].j;
				C[k].v = B[j].v;
				k++;
				j++;
			}
			else{ 
			// row & col both equal
				if(A[i].v + B[j].v != 0){
					C[k].i = A[i].i;
					C[k].j = A[i].j;
					C[k].v = A[i].v + B[j].v;
					k++;	
				}
				i++;
				j++;
			}
		}
	}
	
	if(i == N1 && j != N2){
		while(j < N2){
			C[k].i = B[j].i;
			C[k].j = B[j].j;
			C[k].v = B[j].v;
			k++;
			j++;
		}
	}
	if(j == N2 && i != N1){
		while( i < N1){
			C[k].i = A[i].i;
			C[k].j = A[i].j;
			C[k].v = A[i].v;
			k++;
			i++;
		}
	}

	if(k == 0){
		printf("%d", k);    //no '\n'
	}
	else{
		printf("%d\n", k);
		for(i = 0; i < k-1; i++){
			printf("%d %d %d\n", C[i].i, C[i].j, C[i].v);
		}
		printf("%d %d %d", C[i].i, C[i].j, C[i].v);
	}
	return 0;
}