From c26b78850300acdfb3bc5cf0797f9e9068ccc7fd Mon Sep 17 00:00:00 2001 From: Janita Willumsen Date: Mon, 4 Sep 2023 12:27:14 +0200 Subject: [PATCH] Implemented general algorithm --- src/simpleFile.cpp | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/simpleFile.cpp b/src/simpleFile.cpp index 9d1991a..1b2d600 100644 --- a/src/simpleFile.cpp +++ b/src/simpleFile.cpp @@ -7,7 +7,21 @@ arma::vec* general_algorithm( arma::vec* g_vec ) { + int n = main_diag->n_elem; + double d; + for (int i = 1; i < n; i++) { + d = (*sub_diag)(i-1) / (*main_diag)(i-1); + (*main_diag)(i) -= d*(*sup_diag)(i-1); + (*g_vec)(i) -= d*(*g_vec)(i-1); + } + + (*g_vec)(n-1) /= (*main_diag)(n-1); + + for (int i = n-2; i >= 0; i--) { + (*g_vec)(i) = ((*g_vec)(i) - (*sup_diag)(i) * (*g_vec)(i+1)) / (*main_diag)(i); + } + return g_vec; }