[C/C++] 단위행렬, 영행렬

단위 행렬

: 주대각선 원소가 모두 1이며 나머지 원소는 모두 0인 정사각 행렬

(정사각 행렬은 행과 열의 크기가 같은 행렬)

영 행렬

: 모든 원소가 0인 행렬 (행렬 덧셈의 항등원)

---

C++로 구현하기

사실 단위행렬은 정사각행렬이어야 하지만 정사각행렬이 아니어도 처리하도록 코드를 짰다.

기존 생성자 함수를 오버로딩해 type이라는 인자를 하나 더 받는 경우를 추가했다.

type 인자 값이 0이면 해당 행렬은 영행렬이고, 1이면 단위행렬이다.

두 함수의 경우 값을 입력할 필요가 없기 때문에 입력을 받지 않고 자동으로 값을 할당한다.

#include <iostream>

#include "matrix.h"

using namespace std;

int main() {
    // 영행렬
    Matrix m1(2, 3, 0);
    // 단위 행렬
    Matrix m2(4, 4, 1);

    m1.print();
    m2.print();

    return 0;
}

Matrix 클래스의 수정 부분

...
class Matrix {
    int **mat;
    int row;
    int col;

public:
    ...
    // 생성자(단위행렬, 영행렬)
    // type이 0이면 영행렬, 1이면 단위행렬
    Matrix(int row, int col, int type) : row(row), col(col) {
        mat = (int **)malloc(sizeof(int *) * row);
        for (int i = 0; i < row; i++)
            mat[i] = (int *)malloc(sizeof(int) * col);

        printf("[ %d X %d %s Matrix ]\n", row, col, (type ? "Identity" : "Zero"));
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (type) {
                    if (i == j) mat[i][j] = 1;
                    else mat[i][j] = 0;
                }
                else {
                    mat[i][j] = 0;
                }
            }
        }
        printf("\n");
    }
    ...
};
...