PTA 6-28稀疏矩阵求和

本题实现稀疏矩阵的求和运算。

函数接口定义：

int add_mat(elem a[],int t1,elem b[],int t2, elem c[]);//求a+b

其中 t1 和 t2 表示稀疏矩阵a的长度和稀疏矩阵b的长度，函数返回a+b的长度。

裁判测试程序样例：

#include <stdio.h>
#define M 10  //行
#define N 10  //列
typedef  struct 
{    int  row, col;     //行号、列号
int    val;
}elem;

void input(elem a[],int m);
void show_mat(elem a[],int n);
int add_mat(elem a[],int t1,elem b[],int t2, elem c[]);

void input(elem a[],int m)
{
int i;
for(i=0;i<m;i++)
scanf("%d%d%d",&a[i].row,&a[i].col,&a[i].val);
}

void show_mat(elem a[],int n)//输出
{
    int i;
    for(i=0;i<n;i++)
        printf("%d %d %d\n",a[i].row,a[i].col,a[i].val);
}


int main()
{
    int t,m,n;
    elem a[M*N],b[N*M],c[N*M];
    scanf("%d%d",&m,&n);
    input(a,m);
    input(b,n);
    t=add_mat(a,m,b,n,c);//c=a+b
    show_mat(c,t);
    return 0;
}

/* 请在这里填写答案 */

输入样例：

第一行为矩阵行和列，接下来为两个矩阵

3 2
0 0 1
0 1 3
2 1 2
0 1 -3
1 1 2

输出样例：

0 0 1
1 1 2
2 1 2

 代码实现：

int add_mat(elem a[],int t1,elem b[],int t2, elem c[]){
    int ai = 0, bi = 0, ci = 0;
    int sum;

    while (ai < t1 && bi < t2) {
        // 如果a的元素在b的元素之前（0/1）
        if (a[ai].row < b[bi].row || (a[ai].row == b[bi].row && a[ai].col < b[bi].col)) {  
            c[ci++] = a[ai++]; // a到c
        }
        // 如果b的元素在a的元素之前（1/0）
        else if (a[ai].row > b[bi].row || (a[ai].row == b[bi].row && a[ai].col > b[bi].col)) {  
            c[ci++] = b[bi++]; // b到c
        }
        // 如果两个元素在同一位置
        else {
            sum = a[ai].val + b[bi].val;
            if (sum != 0) {
                c[ci].row = a[ai].row;
                c[ci].col = a[ai].col;
                c[ci].val = sum;
                ci++;
            }
            ai++; 
            bi++;
        }
    }
    while (ai < t1) {
        c[ci++] = a[ai++];
    }  
    while (bi < t2) {
        c[ci++] = b[bi++];
    }
    return ci;
}//拿捏