第九周项目3——稀疏矩阵的三元组表示的实现及应用(1)

  1. /*  
  2. * Copyright (c)2016,烟台大学计算机与控制工程学院  
  3. * All rights reserved.  
  4. * 文件名称:wu.cpp  
  5. * 作    者:武昊  
  6. * 完成日期:2016年10月27日  
  7. * 版 本 号:v1.0   
  8. *问题描述:建立稀疏矩阵三元组表示的算法库,包括:  
  9. ① 头文tup.h,定义数据类型,声明函数;  
  10. ② 源文件tup.cpp,实现稀疏矩阵三元组表示的基本运算,主要算法包括: 
  11.  
  12. void CreatMat(TSMatrix &t,ElemType A[M][N]);  //从一个二维稀疏矩阵创建其三元组表示 
  13. bool Value(TSMatrix &t,ElemType x,int i,int j);  //三元组元素赋值 
  14. bool Assign(TSMatrix t,ElemType &x,int i,int j);  //将指定位置的元素值赋给变量 
  15. void DispMat(TSMatrix t); //输出三元组 
  16. void TranTat(TSMatrix t,TSMatrix &tb);//矩阵转置 
  17.  
  18. ③ 设计main函数,测试上面实现的算法  
  19. *输入描述:无  
  20. *程序输出:测试结果  
  21. */    
tup.h:
  1. #ifndef TUP_H_INCLUDED  
  2. #define TUP_H_INCLUDED  
  3.   
  4. #define M 6  
  5. #define N 7  
  6. #define MaxSize  100         //矩阵中非零元素最多个数  
  7. typedef int ElemType;  
  8.   
  9. typedef struct  
  10. {  
  11.     int r;                  //行号  
  12.     int c;                  //列号  
  13.     ElemType d;             //元素值  
  14. } TupNode;                  //三元组定义  
  15.   
  16. typedef struct  
  17. {  
  18.     int rows;               //行数  
  19.     int cols;               //列数  
  20.     int nums;               //非零元素个数  
  21.     TupNode data[MaxSize];  
  22. } TSMatrix;                 //三元组顺序表定义  
  23.   
  24. void CreatMat(TSMatrix &t,ElemType A[M][N]);  //从一个二维稀疏矩阵创建其三元组表示  
  25. bool Value(TSMatrix &t,ElemType x,int i,int j);  //三元组元素赋值  
  26. bool Assign(TSMatrix t,ElemType &x,int i,int j); //将指定位置的元素值赋给变量  
  27. void DispMat(TSMatrix t);//输出三元组  
  28. void TranTat(TSMatrix t,TSMatrix &tb);//矩阵转置  
  29.   
  30. #endif // TUP_H_INCLUDED  
tup.cpp:
  1. #include "stdio.h"  
  2. #include "glist.h"  
  3.   
  4. void CreatMat(TSMatrix &t,ElemType A[M][N])  //从一个二维稀疏矩阵创建其三元组表示  
  5. {  
  6.     int i,j;  
  7.     t.rows=M;  
  8.     t.cols=N;  
  9.     t.nums=0;  
  10.     for (i=0; i<M; i++)  
  11.     {  
  12.         for (j=0; j<N; j++)  
  13.             if (A[i][j]!=0)     //只存储非零元素  
  14.             {  
  15.                 t.data[t.nums].r=i;  
  16.                 t.data[t.nums].c=j;  
  17.                 t.data[t.nums].d=A[i][j];  
  18.                 t.nums++;  
  19.             }  
  20.     }  
  21. }  
  22.   
  23. bool Value(TSMatrix &t,ElemType x,int i,int j)  //三元组元素赋值  
  24. {  
  25.     int k=0,k1;  
  26.     if (i>=t.rows || j>=t.cols)  
  27.         return false;               //失败时返回false  
  28.     while (k<t.nums && i>t.data[k].r) k++;                  //查找行  
  29.     while (k<t.nums && i==t.data[k].r && j>t.data[k].c) k++;//查找列  
  30.     if (t.data[k].r==i && t.data[k].c==j)   //存在这样的元素  
  31.         t.data[k].d=x;  
  32.     else                                    //不存在这样的元素时插入一个元素  
  33.     {  
  34.         for (k1=t.nums-1; k1>=k; k1--)  
  35.         {  
  36.             t.data[k1+1].r=t.data[k1].r;  
  37.             t.data[k1+1].c=t.data[k1].c;  
  38.             t.data[k1+1].d=t.data[k1].d;  
  39.         }  
  40.         t.data[k].r=i;  
  41.         t.data[k].c=j;  
  42.         t.data[k].d=x;  
  43.         t.nums++;  
  44.     }  
  45.     return true;                        //成功时返回true  
  46. }  
  47.   
  48. bool Assign(TSMatrix t,ElemType &x,int i,int j)  //将指定位置的元素值赋给变量  
  49. {  
  50.     int k=0;  
  51.     if (i>=t.rows || j>=t.cols)  
  52.         return false;           //失败时返回false  
  53.     while (k<t.nums && i>t.data[k].r) k++;                  //查找行  
  54.     while (k<t.nums && i==t.data[k].r && j>t.data[k].c) k++;//查找列  
  55.     if (t.data[k].r==i && t.data[k].c==j)  
  56.         x=t.data[k].d;  
  57.     else  
  58.         x=0;                //在三元组中没有找到表示是零元素  
  59.     return true;            //成功时返回true  
  60. }  
  61.   
  62. void DispMat(TSMatrix t)        //输出三元组  
  63. {  
  64.     int i;  
  65.     if (t.nums<=0)          //没有非零元素时返回  
  66.         return;  
  67.     printf("\t%d\t%d\t%d\n",t.rows,t.cols,t.nums);  
  68.     printf("\t------------------\n");  
  69.     for (i=0; i<t.nums; i++)  
  70.         printf("\t%d\t%d\t%d\n",t.data[i].r,t.data[i].c,t.data[i].d);  
  71. }  
  72.   
  73. void TranTat(TSMatrix t,TSMatrix &tb)       //矩阵转置  
  74. {  
  75.     int p,q=0,v;                    //q为tb.data的下标  
  76.     tb.rows=t.cols;  
  77.     tb.cols=t.rows;  
  78.     tb.nums=t.nums;  
  79.     if (t.nums!=0)                  //当存在非零元素时执行转置  
  80.     {  
  81.         for (v=0; v<t.cols; v++)        //tb.data[q]中的记录以c域的次序排列  
  82.             for (p=0; p<t.nums; p++)    //p为t.data的下标  
  83.                 if (t.data[p].c==v)  
  84.                 {  
  85.                     tb.data[q].r=t.data[p].c;  
  86.                     tb.data[q].c=t.data[p].r;  
  87.                     tb.data[q].d=t.data[p].d;  
  88.                     q++;  
  89.                 }  
  90.     }  
  91. }  
main.cpp:
  1. #include "tup.h"  
  2. int main()  
  3. {  
  4.     TSMatrix t,tb;  
  5.     int x,y=10;  
  6.     int A[6][7]=  
  7.     {  
  8.         {0,0,1,0,0,0,0},  
  9.         {0,2,0,0,0,0,0},  
  10.         {3,0,0,0,0,0,0},  
  11.         {0,0,0,5,0,0,0},  
  12.         {0,0,0,0,6,0,0},  
  13.         {0,0,0,0,0,7,4}  
  14.     };  
  15.     CreatMat(t,A);  
  16.     printf("b:\n");  
  17.     DispMat(t);  
  18.     if (Assign(t,x,2,5)==true)  //调用时返回true  
  19.         printf("Assign(t,x,2,5)=>x=%d\n",x);  
  20.     else  //调用时返回false  
  21.         printf("Assign(t,x,2,5)=>参数错误\n");  
  22.     Value(t,y,2,5);  
  23.     printf("执行Value(t,10,2,5)\n");  
  24.     if (Assign(t,x,2,5)==true)  //调用时返回true  
  25.         printf("Assign(t,x,2,5)=>x=%d\n",x);  
  26.     else  //调用时返回false  
  27.         printf("Assign(t,x,2,5)=>参数错误\n");  
  28.     printf("b:\n");  
  29.     DispMat(t);  
  30.     TranTat(t,tb);  
  31.     printf("矩阵转置tb:\n");  
  32.     DispMat(tb);  
  33.     return 0;  
  34. }  


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
稀疏矩阵是指其中大部分元素为零的矩阵。由于大多数矩阵都是稠密的,即大多数元素都不为零,因此通常情况下,我们使用一个二维数组来表示一个矩阵。但是,对于稀疏矩阵来说,这种方法会造成很大的浪费,因为大量的空间被用来存储零元素。为了解决这个问题,我们可以使用稀疏矩阵三元表示法。 稀疏矩阵三元表示法是将稀疏矩阵中每个非零元素的行、列和值存储在一个三元组中。其数据结构如下所示: ``` struct Triple{ int row, col; double value; }; ``` 其中,row表示非零元素所在的行,col表示非零元素所在的列,value表示非零元素的值。我们可以使用一个数组来存储所有的非零元素,这个数组就是稀疏矩阵三元组。 稀疏矩阵三元表示法的优点是它可以节省存储空间,缺点是它不方便进行矩阵运算。因此,在进行矩阵运算时,我们需要将稀疏矩阵转换成其他更方便进行矩阵运算的表示方法,如压缩矩阵和坐标矩阵等。 对于稀疏矩阵的求解,可以使用稀疏矩阵三元表示法结合三元组高斯消元算法来进行求解。三元组高斯消元算法是一种针对稀疏矩阵的高斯消元算法,其基本思想是将矩阵化为上三角矩阵或下三角矩阵,然后通过回代或者前代求解方程。由于矩阵中大部分元素为零,因此在进行高斯消元时,我们只需要考虑非零元素,这样可以大大提高计算效率。 三元组高斯消元算法的基本步骤如下: 1. 将稀疏矩阵转换成三元表示法; 2. 对三元组按照行和列的顺序进行排序; 3. 从第一个非零元素开始,进行高斯消元,将矩阵化为上三角矩阵或下三角矩阵; 4. 通过回代或者前代求解方程。 具体实现可以参考以下代码: ``` void SparseTripletGaussElimination(SparseTriplet& triplet, vector<double>& b) { int n = triplet.rows; vector<Triple> A(triplet.data, triplet.data + triplet.num); sort(A.begin(), A.end(), [](const Triple& a, const Triple& b){ return a.row < b.row || (a.row == b.row && a.col < b.col); }); vector<int> row(n+1), col(triplet.num), diag(n); vector<double> val(triplet.num); for (int i = 0; i < triplet.num; i++) { row[A[i].row]++; } for (int i = 1; i <= n; i++) { row[i] += row[i-1]; } for (int i = 0; i < triplet.num; i++) { int r = A[i].row, c = A[i].col; double v = A[i].value; int k = row[r]++; // 获取 r 行中下一个非零元素的位置 col[k] = c; val[k] = v; if (r == c) { diag[r] = k; // 记录对角线元素的位置 } } for (int k = 0; k < n-1; k++) { if (val[diag[k]] == 0) { // 对角线元素为零,无法消元 throw runtime_error("zero pivot encountered"); } for (int i = diag[k]+1; i < row[k+1]; i++) { int r = col[i]; double factor = val[i] / val[diag[k]]; for (int j = diag[k]+1; j < row[k+1]; j++) { if (col[j] == r) { val[j] -= factor * val[diag[k]]; } } b[r] -= factor * b[k]; } } if (val[diag[n-1]] == 0) { // 对角线元素为零,无法消元 throw runtime_error("zero pivot encountered"); } for (int k = n-1; k >= 0; k--) { double sum = 0; for (int i = diag[k]+1; i < row[k+1]; i++) { sum += val[i] * b[col[i]]; } b[k] = (b[k] - sum) / val[diag[k]]; } } ``` 其中,SparseTriplet是稀疏矩阵三元表示法的数据结构,b是待求解的方程的右侧向量。在实现中,我们首先将三元组按照行和列的顺序进行排序,然后将其转换成压缩矩阵的形式,接着进行高斯消元,并通过回代或者前代求解方程。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值