【刘庆源码共享】稀疏线性系统求解算法MGMRES(m) 之矩阵类(C++)

最新推荐文章于 2023-04-19 16:06:49 发布

lq_0402

最新推荐文章于 2023-04-19 16:06:49 发布

阅读量1k

点赞数

文章标签： c++ 算法 matrix vector delete string

本文链接：https://blog.csdn.net/lq_0402/article/details/6007719

版权

/*
* Copyright (c) 2009 湖南师范大学数计院一心飞翔项目组
* All Right Reserved
*
* 文件名：matrix.h
* 摘要：定义矩阵类，包括矩阵的相关信息和方法
*
* 作者：刘庆
* 完成日期：2009年2月28日
*
*/

#ifndef _MATRIX_ // 防止matrix.h被重复引用
#define _MATRIX_

#include <string> // 引用标准库的头文件
#include <iostream>
#include "define.h" // 引用非标准库的头文件
#include "Number.h"

using namespace std;

/* 元素类，包含值及索引 */
class Point
{
public:
Double value; /* 数值 */
long index; /* 索引值 */
};

/* 结点类，包含矩阵所有元素的信息 */
class Node
{                    /* MCRF存储法 */
protected:
long length;              /* point的长度 */
public:
Point* point;             /* 结点信息 */

Node();                     /* 默认构造函数 */
Node(const Node& node);        /* 拷贝构造函数 */
void Delete();                           /* 释放point的空间 */
~Node();                                  /* 析构函数 */
Point& operator[](const long index) const; /* 重载[]，判断参数索引值是否越界 */
Node& operator=(const Node& node);       /* 重载等于号，创建一个与参数一样Node对象 */
void GetSpace(const long len);                  /* 替point申请空间 */
long Length() const;                                   /* 得到point的长度 */
};

/* 矩阵类包含矩阵的总行数、总列数、总的非零元素个数，以及矩阵所有元素的信息 */
class Matrix
{
protected:
long total_elem;           /* 总非零元素个数 */
long total_ln;               /* 总行数 */
long total_col;              /* 总列数 */
public:
Node data;                   /* 组合结点类，以表示矩阵元素信息 */

public:
Matrix();                       /* 默认构造函数 */
Matrix(const Matrix &matrix);     /* 拷贝构造函数 */
~Matrix();                                 /* 析构函数 */
void InitZero();                          /* 置0函数，将三个值置为0 */
void SetTotal_elem(long totalElem);  /* 替total_elem赋值 */
long GetTotal_elem();                        /* 返回total_elem */
void SetTotal_ln(long totalLn);          /* 替total_ln赋值 */
long GetTotal_ln();                            /* 返回total_ln */
void SetTotal_col(long totalCol);       /* 替total_col赋值 */
long GetTotal_col();                          /* 返回total_col */

void Init(Double value);                    /* 初始化矩阵信息 index赋值为-1，value赋值为参数值 */
void Display(char *str = 0) const;     /* 显示矩阵信息，参数为提示信息 */
void DisplayAsMatrix(char *str = 0, int zero = 1) const;       /* 以矩阵的形式输出矩阵, zero == 1 输出0元素，否则不输出 */
void QuickSortLn(long *ln,const long low,const long high); /* 用快速排序法将行标排序 */
long QSPartitionLn(long *ln,long low,long high);                /* 实现行标的分区 */
void QuickSortCol(const long low, const long high);           /* 用快速排序法将列标排序 */
long QSPartitionCol(long low,long high);                             /* 实现列标的分区 */
long QuickLocate(const long ln, const long col)const;          /* 用折半查找，搜索[ln,col]位置的元素，不为零则返回该值的位置，否则返回-1; */

void LnTransformMultiply(const long base,const Double& gene); /* 矩阵行初等变换第base行乘以gene */
void LnTransfromAdd(const long base,const long ln);                 /* 矩阵行初等变换第base行加上第ln行 */
Matrix& Transpose();                                              /* 矩阵转置 */
void ToBeIdentityMatrix();                                       /* 单位化一个矩阵 */
void ToBeZeroMatrix();                                           /* 将当前矩阵化为0矩阵 */
void ContraryMatrix();                                             /* 求矩阵的逆返回当前矩阵的逆矩阵 */
long GetMatrixRank() const;                                   /* 求矩阵的秩 */
long GetExtendMatrixRank(const Matrix& b) const; /* 求增广矩阵的秩增光矩阵为 A|b */
int IsEMRequalMR(const Matrix& b) const;               /* 判断增广矩阵的秩和原矩阵的秩是否一致 */
Matrix GetVector(const long ln) const;                     /* 在矩阵中提取ln行作为一个矩阵 */
Matrix GetColVector(const long col) const;              /* 在矩阵中提取col列作为一个矩阵 */
Double GetModulus() const;                                    /* 求矩阵的范数(摸) */
Double GetDotMetrix(const Matrix& matrix)const;   /* 返回两矩阵的点积 */
int AddLine(const Matrix &vector);                          /* 向矩阵中添加1行 */
int AddCol(const Matrix& verctor);                           /* 向矩阵中添加1列 */
Matrix& operator *(const Matrix &genMatrix);         /* 两矩阵相乘返回当前矩阵 */
Matrix& operator +(const Matrix &genMatrix);         /* 两矩阵相加返回当前矩阵 */
Matrix& operator *(const Double& gen);                 /* 矩阵的数乘返回当前矩阵 */
Matrix& operator /(const Double& gen);                 /* 矩阵除以一个数返回当前矩阵 */
Matrix& operator =(const Matrix &matrix);               /* 赋值运算，创建一个与参数一样的矩阵返回当前矩阵 */
Matrix& operator -(const Matrix &genMatrix);          /* 两矩阵相减返回当前矩阵 */
friend ostream& operator<<(ostream& os, const Matrix& matrix);               /* 输出矩阵信息 */
Matrix& PosMultiply(const long ln, const long col, const Double& gen);        /* 第ln,col位置的元素乘以gen */
Matrix& PosSetValue(const long ln, const long col, const Double& value);   /* 替ln,col位置赋值 */
Matrix& DeleteElement(const long ln, const long col);                                  /* 删除ln,col位置的元素 */
Matrix& AddElement(const long ln, const long col, const Double& value);   /* 增加ln和col位置的元素 */
void OutFile(char* fileName)const;                            /* 将矩阵的信息输出到fileName文件中 */
static Matrix GetE(const long total_ln, const long j);  /* 得到一个total_ln行，第j行为1其他为0的矩阵 */
static Matrix GetE(const long ln);                               /* 得到一个全为1的矩阵 */
};