FYS3150/Project-1
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implemented general algorithm

Browse Source
This commit is contained in:
Janita Willumsen 2023-09-04 12:27:14 +02:00
parent 6cff40e678
commit c26b788503
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/simpleFile.cpp
View File

@ -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;
}