第8周 项目4-稀疏矩阵的三元组表示的实现及应用(2)

/* * Copyright(c) 2017,烟台大学计算机学院 * All rights reserved. * 文件名称:cpp1. * 作    者:陈军正 * 完成日期:2017 年 12 月 8 日 * 版 本 号:v1.0 * * 问题描述:采用三元组存储稀疏矩阵，设计两个稀疏矩阵相加的运算算法 提示1：两个行数、列数相同的矩阵可以相加 提示2：充分利用已经建立好的算法库解决问题 提示3：教材例6.3已经给出两个稀疏矩阵相加的运算的算法，但未利用基本运算。请比较这两种方案 * 输入描述:无需输入 * 程序输出:实现各种算法的函数的测试结果 */  

解一：

#include <stdio.h>  #include "tup.h"  bool MatAdd(TSMatrix a,TSMatrix b,TSMatrix &c)  {      int i,j;      ElemType va,vb,vc;      if (a.rows!=b.rows || a.cols!=b.cols)          return false;                        //行数或列数不等时不能进行相加运算      c.rows=a.rows;      c.cols=a.cols;       //c的行列数与a的相同      c.nums=0;      for(i=0; i<M; i++)          for(j=0; j<N; j++)          {              Assign(a,va,i,j);              Assign(b,vb,i,j);              vc=va+vb;              if(vc)                  Value(c,vc,i,j);          }      return true;  }    int main()  {      TSMatrix ta,tb,tc;      int A[M][N]=      {          {0,0,1,0,0,0,0},          {0,2,0,0,0,0,0},          {3,0,0,0,0,0,0},          {0,0,0,5,0,0,0},          {0,0,0,0,6,0,0},          {0,0,0,0,0,7,4}      };      int B[M][N]=      {          {0,0,10,0,0,0,0},          {0,0,0,20,0,0,0},          {0,0,0,0,0,0,0},          {0,0,0,50,0,0,0},          {0,0,20,0,0,0,0},          {0,0,0,10,0,0,4}      };      CreatMat(ta,A);      CreatMat(tb,B);      printf("A:\n");      DispMat(ta);      printf("B:\n");      DispMat(tb);      if(MatAdd(ta, tb, tc))      {          printf("A+B:\n");          DispMat(tc);      }      else      {          printf("相加失败\n");      }      return 0;  }  

解二：

#include <stdio.h>  #include "tup.h"  bool MatAdd(TSMatrix a,TSMatrix b,TSMatrix &c)  {      int i=0,j=0,k=0;      ElemType v;      if (a.rows!=b.rows || a.cols!=b.cols)          return 0;        //行数或列数不等时不能进行相加运算      c.rows=a.rows;      c.cols=a.cols;       //c的行列数与a的相同      while (i<a.nums && j<b.nums)         //处理a和b中的每个元素      {          if (a.data[i].r==b.data[j].r)    //行号相等时          {              if(a.data[i].c<b.data[j].c)  //a元素的列号小于b元素的列号              {                  c.data[k].r=a.data[i].r;//将a元素添加到c中                  c.data[k].c=a.data[i].c;                  c.data[k].d=a.data[i].d;                  k++;                  i++;              }              else if (a.data[i].c>b.data[j].c)//a元素的列号大于b元素的列号              {                  c.data[k].r=b.data[j].r;      //将b元素添加到c中                  c.data[k].c=b.data[j].c;                  c.data[k].d=b.data[j].d;                  k++;                  j++;              }              else                    //a元素的列号等于b元素的列号              {                  v=a.data[i].d+b.data[j].d;                  if (v!=0)           //只将不为0的结果添加到c中                  {                      c.data[k].r=a.data[i].r;                      c.data[k].c=a.data[i].c;                      c.data[k].d=v;                      k++;                  }                  i++;                  j++;              }          }          else if (a.data[i].r<b.data[j].r) //a元素的行号小于b元素的行号          {              c.data[k].r=a.data[i].r;      //将a元素添加到c中              c.data[k].c=a.data[i].c;              c.data[k].d=a.data[i].d;              k++;              i++;          }          else                              //a元素的行号大于b元素的行号          {              c.data[k].r=b.data[j].r;      //将b元素添加到c中              c.data[k].c=b.data[j].c;              c.data[k].d=b.data[j].d;              k++;              j++;          }          c.nums=k;      }      return true;  }    int main()  {      TSMatrix ta,tb,tc;      int A[M][N]=      {          {0,0,1,0,0,0,0},          {0,2,0,0,0,0,0},          {3,0,0,0,0,0,0},          {0,0,0,5,0,0,0},          {0,0,0,0,6,0,0},          {0,0,0,0,0,7,4}      };      int B[M][N]=      {          {0,0,10,0,0,0,0},          {0,0,0,20,0,0,0},          {0,0,0,0,0,0,0},          {0,0,0,50,0,0,0},          {0,0,20,0,0,0,0},          {0,0,0,10,0,0,4}      };      CreatMat(ta,A);      CreatMat(tb,B);      printf("A:\n");      DispMat(ta);      printf("B:\n");      DispMat(tb);      if(MatAdd(ta, tb, tc))      {          printf("A+B:\n");          DispMat(tc);      }      else      {          printf("相加失败\n");      }      return 0;  }  

运行结果：