템플릿과 역행렬 1. template을 써보자

이번 떡밥은 템플릿 공부도 하고 수학 공부도 할겸 템플릿을 이용해서 n차행렬의 역행렬을 구해보는것이다. c++를 좀 써본 사람이라면 template<>어쩌구 하는 문법은 알거라고 생각하고 본론으로 들어간다.

이런 행렬을

이렇게 묶어주고

묶어서 생긴 새로운 행렬들을 A,B,C,D라 하면

이렇게 표현할수 있다. (행렬들의 행렬에 대해서도 덧셈, 뺄셈, 곱셈, 실수배가 똑같이 정의된다.) 이런 방법을 재귀적으로 사용한다면 2^n꼴의 행렬을 2*2행렬의 형태로 표현가능하다.

이 사실을 바탕으로 matrix 클래스를 짜본다.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37

template<int dim> class matrix { protected:     enum{halfdim=dim/2};     matrix<halfdim> a,b,c,d; public:     matrix<dim>()     {     }     matrix<dim>(matrix<halfdim> _a, matrix<halfdim> _b,         matrix<halfdim> _c, matrix<halfdim> _d)     {         a=_a;         b=_b;         c=_c;         d=_d;     } };   template<> class matrix<2> { protected:     double a,b,c,d; public:     matrix<2>()     {     }     matrix<2>(double _a, double _b, double _c, double _d)     {         a=_a;         b=_b;         c=_c;         d=_d;     } };

dim=2일 경우에는 double a,b,c,d를 가지게하고, dim!=2일때는 matrix<dim/2> a,b,c,d를 가지게 했다. 물론 dim이 2^n일 경우에만 정상적으로 작동한다.
음, 쉽군. 이제 연산자를 정의한다. +,-,*는 매우 쉽게 짤 수 있다.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

matrix<dim> operator+(matrix<dim> &t) {     return matrix<dim>(a+t.a, b+t.b, c+t.c, d+t.d); } matrix<dim> operator-(matrix<dim> &t) {     return matrix<dim>(a-t.a, b-t.b, c-t.c, d-t.d); } matrix<dim> operator-() {     return matrix<dim>(-a, -b, -c, -d); } matrix<dim> operator*(matrix<dim> &t) {     return matrix<dim>(a*t.a + b*t.c, a*t.b + b*t.d,          c*t.a + d*t.c, c*t.b + d*t.d); }

물론 이 코드들은 template<int dim> class matrix 안에 들어가야겠다. 귀찮겠지만 특수화한 템플릿 클래스 matrix<2>에 대해서도 마찬가지로 정의해줘야한다.

이제 자주 쓰이는 영행렬을 정적변수로 선언한다.

1 2 3 4

static matrix<dim> zero; //이 놈은 template<int dim> class matrix안에 static matrix<2> zero; //이 놈은 class matrix<2>안에 들어가야한다.

그리고 외부에 다음과 같이 정의해야한다.

1 2 3 4 5 6 7

template<int dim> matrix<dim> matrix<dim>::zero=matrix<dim> (matrix<halfdim>::zero, matrix<halfdim>::zero,  matrix<halfdim>::zero, matrix<halfdim>::zero); //이 정의는 템플릿과 같은 헤더에 들어가있어야하지만, matrix<2> matrix<2>::zero=matrix<2>(0.0, 0.0, 0.0, 0.0); //이 정의는 소스파일에 들어가있어야 중복정의되었다는 에러가 안 뜬다.

이제 역행렬 구하는 것만이 남았는데, 쉬운것부터 하자. matrix<2>먼저 구현해보면

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

bool has_inverse() {     return a*d!=b*c; } double det() {     return a*d-b*c; } matrix<2> inverse() {     double dt=det();     if(dt!=0)     {         return matrix<2>(d/dt, -b/dt, -c/dt, a/dt);     }     return zero; }

이 코드들도 역시 class matrix<2>안에 들어가야겠다.
 
행렬들의 행렬의 역행렬을 구하는 것은 실수들의 행렬의 역행렬을 구하는것보다 까다롭다. 그래서 그 부분은 다음 글로 넘겼다.