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

Merge branch '7-solve-problem-7' of github.uio.no:FYS3150-G2-2023/Project-1 into 7-solve-problem-7

Browse Source
This commit is contained in:
Cory Balaton 2023-09-04 12:35:56 +02:00
commit b7195f8c17
No known key found for this signature in database
GPG Key ID: 3E5FCEBFD80F432B
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

@ -12,7 +12,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;
}