Не работает перестановка столбцов

This commit is contained in:
AZEN-SGG 2025-04-02 01:36:54 +03:00
parent b4c7f2d872
commit 0c5beccc85

View file

@ -2,32 +2,55 @@
#include "io_status.h" #include "io_status.h"
#include <float.h> #include <float.h>
#include <math.h> #include <math.h>
#include "array_io.h"
#include <stdio.h>
// c - changes in rows // c - changes in rows
int t14_solve(int n, double * restrict A, double * restrict X, int * restrict c) int t14_solve(int n, double * restrict A, double * restrict X, int * restrict c)
{ {
// Проходимся по главным минорам
for (int k = 0; k < n; ++k) { for (int k = 0; k < n; ++k) {
double maximum = -1.; double maximum = -1.;
int max_i = 0, max_j = 0; int max_i = 0, max_j = 0;
//printf("\n--------- K = %d ---------\n", k);
// Ищем максимальный элемент минора
#pragma omp parallel
{
double local_max = -1.;
int loc_i = 0, loc_j = 0;
#pragma omp parallel for collapse(2) reduction(max:maximum) #pragma omp for collapse(2) nowait
for (int i = k; i < n; ++i) for (int i = k; i < n; ++i)
for (int j = k; j < n; ++j) for (int j = k; j < n; ++j)
{
double aij = fabs(A[i * n + j]);
#pragma omp critical
{ {
if (aij > maximum) { double aij = fabs(A[i * n + j]);
maximum = aij; if (aij > local_max) {
max_i = i; local_max = aij;
max_j = j; loc_i = i;
loc_j = j;
} }
} }
#pragma omp critical
{
if (local_max > maximum) {
maximum = local_max;
max_i = loc_i;
max_j = loc_j;
}
} }
}
// Если максимальный по модулю элемент равен нулю, значит матрица вырождена
if (fabs(maximum) < DBL_EPSILON) if (fabs(maximum) < DBL_EPSILON)
return SINGULAR; return SINGULAR;
//printf("Maximum = %lf for i = %d, j = %d\n", maximum, max_i, max_j);
// Меняем строки местами, если максимум находится не в k строке
if (max_i != k) if (max_i != k)
{ {
int kn = k*n; int kn = k*n;
@ -56,10 +79,14 @@ int t14_solve(int n, double * restrict A, double * restrict X, int * restrict c)
} }
} }
//print_matrix(A, n, n);
//printf("\n");
// Меняем столбцы местами
if (max_j != k) if (max_j != k)
{ {
int swap_temp = c[max_j]; int swap_temp = c[max_j];
c[max_j] = k; c[max_j] = c[k];
c[k] = swap_temp; c[k] = swap_temp;
#pragma omp simd #pragma omp simd
@ -70,36 +97,44 @@ int t14_solve(int n, double * restrict A, double * restrict X, int * restrict c)
A[in + max_j] = swap; A[in + max_j] = swap;
} }
} }
/*
print_matrix(A, n, n);
printf("\n");
*/
gauss_inverse(n, k, A, X); gauss_inverse(n, k, A, X);
/*
print_matrix(A, n, n);
printf("Inverse matrix:\n");
print_matrix(X, n, n);
*/
} }
gauss_back_substitution(n, A, X); gauss_back_substitution(n, A, X);
// Возвращаем строки назад
for (int k = 0; k < n; ++k) for (int k = 0; k < n; ++k)
{ {
int str_k = k;
int str_i = c[k]; int str_i = c[k];
if (str_i != k) if (str_i != k)
{
#pragma omp parallel for
for (int j = 0; j < n; ++j) for (int j = 0; j < n; ++j)
{ {
int loc_k = k;
int loc_i = str_i;
double elem = X[k*n + j]; double elem = X[k*n + j];
do { do {
X[str_i*n + j] = elem; X[loc_i*n + j] = elem;
elem = X[str_i*n + j]; elem = X[loc_i*n + j];
str_k = str_i; loc_k = loc_i;
str_i = c[str_i]; loc_i = c[loc_i];
c[str_k] = str_k; if (j == n-1)
} while (str_i != k); c[loc_k] = loc_k;
} while (loc_i != k);
X[k*n + j] = elem; X[k*n + j] = elem;
} }
}
} }
return 0; return 0;
@ -112,7 +147,7 @@ void gauss_inverse(const int n, const int k, double * restrict A, double * restr
const double inv_akk = 1./A[kn + k]; const double inv_akk = 1./A[kn + k];
A[kn + k] = 1.; A[kn + k] = 1.;
for (int ij = kn; ij < kn+k; ++ij) for (int ij = kn; ij <= kn+k; ++ij)
{ {
double xij = X[ij]; double xij = X[ij];
if (fabs(xij) > DBL_EPSILON) X[ij] = xij*inv_akk; if (fabs(xij) > DBL_EPSILON) X[ij] = xij*inv_akk;
@ -146,8 +181,10 @@ void gauss_inverse(const int n, const int k, double * restrict A, double * restr
} }
} }
// Обратный ход метода Гаусса
void gauss_back_substitution(const int n, double * restrict A, double * restrict X) void gauss_back_substitution(const int n, double * restrict A, double * restrict X)
{ {
// Идём с последней строки и вычитаем её из последующих
for (int k = n-1; k > 0; --k) for (int k = n-1; k > 0; --k)
{ {
const int kn = k * n; const int kn = k * n;