main
test-branch
| Line | |
|---|
| 1 | /*
|
|---|
| 2 | * Cholesky decomposition;
|
|---|
| 3 | * Code from: https://rosettacode.org/wiki/Cholesky_decomposition#C
|
|---|
| 4 | */
|
|---|
| 5 |
|
|---|
| 6 | #include <stdio.h>
|
|---|
| 7 | #include <stdlib.h>
|
|---|
| 8 | #include <math.h>
|
|---|
| 9 |
|
|---|
| 10 | double *cholesky(double *A, int n) {
|
|---|
| 11 | double *L = (double*)malloc(n*n*sizeof(double));
|
|---|
| 12 |
|
|---|
| 13 | for (int i = 0; i < n*n; i++)
|
|---|
| 14 | L[i] = 0.0;
|
|---|
| 15 |
|
|---|
| 16 | for (int i = 0; i < n; i++)
|
|---|
| 17 | for (int j = 0; j < (i+1); j++) {
|
|---|
| 18 | double sum = 0;
|
|---|
| 19 | for (int k = 0; k < j; k++)
|
|---|
| 20 | sum += L[i * n + k] * L[j * n + k];
|
|---|
| 21 | L[i * n + j] = (i == j) ?
|
|---|
| 22 | sqrt(A[i * n + i] - sum) :
|
|---|
| 23 | (1.0 / L[j * n + j] * (A[i * n + j] - sum));
|
|---|
| 24 | }
|
|---|
| 25 |
|
|---|
| 26 | return L;
|
|---|
| 27 | }
|
|---|
| 28 |
|
|---|
| 29 | void show_matrix(double *A, int n) {
|
|---|
| 30 | for (int i = 0; i < n; i++) {
|
|---|
| 31 | for (int j = 0; j < n; j++)
|
|---|
| 32 | printf("%f ", A[i * n + j]);
|
|---|
| 33 | printf("\n");
|
|---|
| 34 | }
|
|---|
| 35 | }
|
|---|
| 36 |
|
|---|
Note:
See
TracBrowser
for help on using the repository browser.