NOJ数据结构012——以三元组为存储结构实现矩阵相加

àì夳堔傛蜴生んèń 2022-11-17 13:54 131阅读 0赞

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl81MzA2ODI2OQ_size_16_color_FFFFFF_t_70]  
本题练习三元组的应用。  
思路较为简单，即比较两个三元组a、b行数列数后进行排序。  
需要特别注意的是要考虑两个元素相加为0的情况，不能让这样的元素放在c中。

#include <stdio.h>
    #include <stdlib.h>
    
    typedef struct
    {
        
        int row,col;   /*非零元行下标、列下标*/
        int data;      /*非零元数据*/
    }Triple;
    
    typedef struct
    {
        
        Triple *elem;   /*非零元的三元组表*/
        int len;        /*非零元个数*/
    }TSMatrix;
    
    TSMatrix a, b, c;
    
    TSMatrix InitMatrix(int t);          //初始化三元矩阵
    void CreateMatrix(TSMatrix *mat);    //输入矩阵数据
    void AddMatrix(TSMatrix *a, TSMatrix *b, TSMatrix *c); //矩阵相加
    void PrintMatrix(TSMatrix *c);       //打印矩阵
    
    int main()
    {
        
        int t1, t2;
        scanf("%d%d", &t1, &t2);
        a = InitMatrix(t1);
        b = InitMatrix(t2);
        c = InitMatrix(t1 + t2);
        CreateMatrix(&a);
        CreateMatrix(&b);
        AddMatrix(&a, &b, &c);
        PrintMatrix(&c);
        return 0;
    }
    
    TSMatrix InitMatrix(int t)
    {
            //初始化三元矩阵
        TSMatrix Mat;
        Mat.elem = (Triple*) malloc (sizeof(Triple)*t);
        Mat.len = t;
        return Mat;
    }
    
    void CreateMatrix(TSMatrix *mat)
    {
           //输入矩阵数据
        int tmp_row, tmp_col, tmp_data;
        for (int i = 0; i < mat->len; i++) {
        
            scanf("%d%d%d", &tmp_row, &tmp_col, &tmp_data);
            mat->elem[i].row = tmp_row;
            mat->elem[i].col = tmp_col;
            mat->elem[i].data = tmp_data;
        }
    }
    
    void AddMatrix(TSMatrix *a, TSMatrix *b, TSMatrix *c)
    {
           //矩阵相加
        int i = 0, j = 0, k = 0;
        c->len = 0;
        while (i < a->len && j < b->len) {
        
            if (a->elem[i].row < b->elem[j].row) {
         //a的行数小于b的行数
                c->elem[k] = a->elem[i];
                c->len++;
                i++;
                k++;
            }
            else if (a->elem[i].row > b->elem[j].row) {
         //a的行数大于b的行数
                c->elem[k] = b->elem[j];
                c->len++;
                j++;
                k++;
            }
            else if (a->elem[i].row == b->elem[j].row) {
         //a的行数等于b的行数
                if (a->elem[i].col < b->elem[j].col) {
         //a的列数小于b的列数
                    c->elem[k] = a->elem[i];
                    c->len++;
                    i++;
                    k++;
                }
                else if (a->elem[i].col > b->elem[j].col) {
        //a的列数大于b的列数
                    c->elem[k] = b->elem[j];
                    c->len++;
                    j++;
                    k++;
                }
                else if (a->elem[i].col == b->elem[j].col) {
        //a的列数等于b的列数
                    int tmp = a->elem[i].data + b->elem[j].data;
                    if (tmp != 0) {
          //两元素相加不为0
                        c->elem[k].row = a->elem[i].row;
                        c->elem[k].col = a->elem[i].col;
                        c->elem[k].data = a->elem[i].data + b->elem[j].data;
                        c->len++;
                        i++;
                        j++;
                        k++;
                    }
                    else {
        
                        i++;
                        j++;
                    }
                }
            }
        }
        /*当两个三元组有剩余元素时*/
        while(i < a->len) {
        
            c->elem[k] = a->elem[i];
            c->len++;
            k++;
            i++;
        }
        while(j < b->len) {
        
            c->elem[k] = b->elem[j];
            c->len++;
            k++;
            j++;
        }
    }
    
    void PrintMatrix(TSMatrix *c)
    {
          //打印矩阵
        for (int i = 0; i < c->len; i++) {
        
            printf("%d %d %d\n", c->elem[i].row, c->elem[i].col, c->elem[i].data);
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl81MzA2ODI2OQ_size_16_color_FFFFFF_t_70]: /images/20221022/3e78ec68ccd249eab5b2e1caf557b168.png