我想实现strassen的算法,但我要实现矩阵类。但是我使用以下代码得到了分段错误:
class Matrix {
public:
vector < vector <int> > m;
int r;
int c;
Matrix() {;}
Matrix(int _r, int _c) {
r = _r;
c = _c;
m.resize(r);
for (int i = 0; i < r; i++) {
m[i].resize(c);
}
for(int i =0; i < r; i++){
for(int j=0; j< c; j++){
m[i][j] = 0;
}
}
}
friend ostream& operator<<(ostream &out, Matrix &A) {
for(int i =0; i<A.r; i++){
out << endl;
for(int j=0; j<A.c; j++){
out << A.m[i][j] << "\t";
}
}
out<< endl;
return out;
}
Matrix(const Matrix &A) {
c = A.c;
r = A.r;
m.resize(r);
for (int i = 0; i < r; i++) {
m[i].resize(c);
}
for(int i =0; i<r; i++){
for(int j=0; j<c; j++){
m[i][j] = A.m[i][j];
}
}
}
Matrix& operator-= (Matrix &A) {
assert(A.r == r);
assert(A.c == c);
for(int i =0; i<r; i++){
for(int j=0; j<c; j++){
m[i][j] -= A.m[i][j];
}
}
return *this;
}
Matrix& operator- (Matrix &A) {
Matrix C(*this);
return C-=A;
}
Matrix& operator+= (Matrix &A) {
assert(A.r == r);
assert(A.c == c);
for(int i =0; i<r; i++){
for(int j=0; j<c; j++){
m[i][j] += A.m[i][j];
}
}
return *this;
}
Matrix& operator+ (Matrix &A) {
Matrix C (*this);
(C)+=A;
return C;
}
Matrix getBlock(int sR, int eR, int sC, int eC) {
assert(sR > eR);
assert(sC > eC);
Matrix C(eR-sR, eC-sC);
for(int i = 0; i < C.r; i++) {
for(int j=0; j < C.c; j++) {
C.m[i][j] = m[sR+i][sC+j];
}
}
return C;
}
friend void swap(Matrix& first, Matrix& second) {
using std::swap;
swap(first.r, second.r);
swap(first.c, second.c);
swap(first.m, second.m);
}
Matrix& operator=(Matrix other){
return *this;
}
friend Matrix& operator*(const Matrix& A, const Matrix &B) {
assert(A.r == B.c);
Matrix C(A.r, B.c);
for(int i =0; i<C.r; i++){
for(int j=0; j<C.c; j++){
for(int k = 0; k < A.r; k++) {
C.m[i][j] += A.m[i][k] * B.m[k][j];
}
}
}
return C;
}
};
int main (void)
{
Matrix A(2,2), B(2,2);
A.m[0][0] = 1; A.m[0][1] = 2;
A.m[1][0] = 3; A.m[1][1] = 4;
B.m[0][0] = 1; B.m[0][1] = 2;
B.m[1][0] = 3; B.m[1][1] = 4;
Matrix C(2,2);
C =A+B;
cout << C << endl;
return 0;
}
如果我尝试A + = B;它有效...我不明白它与A + B的区别。我试图在从
返回之前打印C.Matrix& operator+ (Matrix &
Matrix C (*this);
(C)+=A;
return C;
}
这是正确的。当代码命中返回时,我的程序崩溃了。我想了解我做错了什么。非常感谢。 的Davide
答案 0 :(得分:3)
在operator+
中,您将返回对本地构造对象的引用。该对象将在函数的出口处被销毁,因此您最终会得到一个悬空引用。与operator*
相同的故事。这是一个关于运算符重载的很好的教程
http://courses.cms.caltech.edu/cs11/material/cpp/donnie/cpp-ops.html
答案 1 :(得分:0)
您不能将局部变量用作l值返回值。
Matrix& operator=(Matrix && other){
this->m = other.m;
return *this;
}
Matrix operator+ (Matrix &A) {
Matrix C (*this);
(C)+=A;
return std::move(C);
}
这可以解决这个问题。
答案 2 :(得分:0)
正如其他人所指出的那样,问题是你返回对局部变量的引用,当该函数退出时会被销毁。
实现operator+
的常用方法是作为非成员函数:
Matrix operator+(Matrix A, Matrix const &B)
{
return A += B;
}
这种方法比成员函数更好,因为它允许在左侧操作数上进行转换。
您的operator-
和operator*
有同样的问题。