diff options
author | Yuqian Yang <crupest@crupest.life> | 2025-02-28 23:13:39 +0800 |
---|---|---|
committer | Yuqian Yang <crupest@crupest.life> | 2025-02-28 23:13:39 +0800 |
commit | dc1f0c4c0096013799416664894c5194dc7e1f52 (patch) | |
tree | 2f5d235f778cd720f4c39ec3e56b77ba6d99f375 /store/works/life/numerical-analysis-lab/GaussianElimination.cpp | |
parent | 7299d424d90b1effb6db69e3476ddd5af72eeba4 (diff) | |
download | crupest-dc1f0c4c0096013799416664894c5194dc7e1f52.tar.gz crupest-dc1f0c4c0096013799416664894c5194dc7e1f52.tar.bz2 crupest-dc1f0c4c0096013799416664894c5194dc7e1f52.zip |
chore(store): move everything to store.
Diffstat (limited to 'store/works/life/numerical-analysis-lab/GaussianElimination.cpp')
-rw-r--r-- | store/works/life/numerical-analysis-lab/GaussianElimination.cpp | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/store/works/life/numerical-analysis-lab/GaussianElimination.cpp b/store/works/life/numerical-analysis-lab/GaussianElimination.cpp new file mode 100644 index 0000000..777eee7 --- /dev/null +++ b/store/works/life/numerical-analysis-lab/GaussianElimination.cpp @@ -0,0 +1,120 @@ +#include <cstdlib> +#include <iomanip> +#include <iostream> + +void PrintMatrix(int n, double **a) { + for (int i = 0; i < n; ++i) { + for (int j = 0; j < n; ++j) { + std::cout << a[i][j] << ' '; + } + std::cout << '\n'; + } +} + +void PrintEquation(int n, double **a, double *b) { + for (int i = 0; i < n; ++i) { + for (int j = 0; j < n; ++j) { + std::cout << a[i][j] << ' '; + } + std::cout << b[i] << '\n'; + } +} + +bool HasSolution(int n, double **a, double *b) { + for (int r = 0; r < n; r++) { + bool all_zero = true; + for (int c = 0; c < n; c++) { + if (a[r][c] != 0) { + all_zero = false; + break; + } + } + + if (all_zero && b[r] != 0) { + return false; + } + } + return true; +} + +void GaussianElimination(int n, double **a, double *b) { + for (int c = 0; c < n; c++) { + int pivot_r = -1; + for (int r = c; r < n; r++) { + if (a[r][c] != 0) { + pivot_r = r; + break; + } + } + + if (pivot_r == -1) { + continue; + } + + double pivot = a[pivot_r][c]; + + for (int r = 0; r < n; r++) { + if (r == pivot_r) { + continue; + } + + double factor = a[r][c] / pivot; + + for (int i = 0; i < n; i++) { + a[r][i] -= factor * a[pivot_r][i]; + b[i] -= factor * b[pivot_r]; + } + } + + for (int i = 0; i < n; i++) { + a[pivot_r][i] /= pivot; + b[pivot_r] /= pivot; + } + } +} + +double GetDiff(int n, double *x, double **a, double *b) { + double diff = 0; + for (int r = 0; r < n; r++) { + for (int c = 0; c < n; c++) { + diff += a[r][c] * x[c]; + } + diff -= b[r]; + } + return diff; +} + +int main() { + std::cout << std::showpos << std::fixed << std::setprecision(3); + + const int n = 6; + double **a = new double *[n]; + double *b = new double[n]; + + for (int i = 0; i < n; i++) { + a[i] = new double[n]; + for (int j = 0; j < n; j++) { + std::cin >> a[i][j]; + } + std::cin >> b[i]; + } + + std::cout << "Equation:\n"; + PrintEquation(n, a, b); + + GaussianElimination(n, a, b); + std::cout << "After Gaussian Elimination:\n"; + PrintEquation(n, a, b); + + if (!HasSolution(n, a, b)) { + std::cout << "No solution!\n"; + } + + if (std::abs(GetDiff(n, b, a, b)) > 1e-6) { + std::cout << "Solution examination failed!\n"; + } else { + std::cout << "Solution examination passed!\n"; + } + + return 0; +} |