Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4d0d8f0d83
|
||
|
|
8498520398
|
||
|
|
5cbb675860
|
||
|
|
ea484363de
|
@@ -1,9 +0,0 @@
|
|||||||
3
|
|
||||||
3 2
|
|
||||||
44 62
|
|
||||||
44 43 30
|
|
||||||
3 34
|
|
||||||
27 63 53
|
|
||||||
14 52 19
|
|
||||||
|
|
||||||
3 3 2 44 62 44 43 30 3 34 27 63 53 14 52 19
|
|
||||||
158
2023207/main.cpp
158
2023207/main.cpp
@@ -1,158 +0,0 @@
|
|||||||
#include <stdio.h>
|
|
||||||
int p = 0;
|
|
||||||
int m = 0, n = 0;
|
|
||||||
double A[5][10000] = {0}, Z[10000] = {0};
|
|
||||||
|
|
||||||
double get_content(int row, int column) {
|
|
||||||
if (row - column > 2 || row - column < -2) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
if (row < 0 || row >= n) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
if (column < 0 || column >= n) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
return A[(row - column + 2)][column];
|
|
||||||
}
|
|
||||||
|
|
||||||
int set_content(int row, int column, double val) {
|
|
||||||
if (row - column > 2 || row - column < -2) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
if (row < 0 || row >= n) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
if (column < 0 || column >= n) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
A[(row - column + 2)][column] = val;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int output() {
|
|
||||||
for (int i = 0; i < p; i++) {
|
|
||||||
for (int j = 0; j < n; j++) {
|
|
||||||
printf("%lf ", get_content(i, j));
|
|
||||||
}
|
|
||||||
printf("\n");
|
|
||||||
}
|
|
||||||
printf("\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int outputZ() {
|
|
||||||
for (int i = 0; i < n; i++) {
|
|
||||||
printf("%lf ", Z[i]);
|
|
||||||
}
|
|
||||||
printf("\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
scanf("%d", &p);
|
|
||||||
scanf("%d %d", &n, &m);
|
|
||||||
// for (int i = 0; i < p / 2; i++) {
|
|
||||||
// for (int j = p / 2 - i; j < n; j++) {
|
|
||||||
// scanf("%lf", &A[i][j]);
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// for (int i = p / 2; i < p; i++) {
|
|
||||||
// for (int j = 0; j < n - (i - p / 2); j++) {
|
|
||||||
// scanf("%lf", &A[i][j]);
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
for (int i = 2 - p / 2; i <= 2; i++) {
|
|
||||||
for (int j = 2 - i; j < n; j++) {
|
|
||||||
scanf("%lf", &A[i][j]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (int i = 3; i <= 2 + p / 2; i++) {
|
|
||||||
for (int j = 0; j < n - (i - 2); j++) {
|
|
||||||
scanf("%lf", &A[i][j]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("Readin\n");
|
|
||||||
// output();
|
|
||||||
|
|
||||||
for (int col = 0; col < n; col++) {
|
|
||||||
double diag_elem = get_content(col, col);
|
|
||||||
// make everything below the diag to 0 row by row
|
|
||||||
for (int row = col + 1; row < n; row++) {
|
|
||||||
double base = get_content(row, col);
|
|
||||||
if (base != 0) {
|
|
||||||
double coefficient = base / diag_elem;
|
|
||||||
set_content(row, col, -coefficient);
|
|
||||||
for (int l = col + 1; l < n; l++) {
|
|
||||||
set_content(row, l, get_content(row, l) - coefficient * get_content(col, l));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("L ready\n");
|
|
||||||
// output();
|
|
||||||
|
|
||||||
for (int row = 0; row < n; row++) {
|
|
||||||
double diag_elem = get_content(row, row);
|
|
||||||
set_content(row, row, 1 / diag_elem);
|
|
||||||
for (int col = row + 1; col < n; col++) {
|
|
||||||
set_content(row, col, -get_content(row, col) / diag_elem);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("All ready\n");
|
|
||||||
// output();
|
|
||||||
|
|
||||||
for (int i = 0; i < m; i++) {
|
|
||||||
// read in one col of z
|
|
||||||
for (int j = 0; j < n; j++) {
|
|
||||||
scanf("%lf", &Z[j]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("Readin\n");
|
|
||||||
// outputZ();
|
|
||||||
|
|
||||||
// calculate Z = L-1 * Z
|
|
||||||
// do this column by column
|
|
||||||
for (int j = 0; j < n; j++) {
|
|
||||||
// diag of U is always 1, z[j] * U[j][j] = z[j], so k doesn't have to equal to j
|
|
||||||
for (int k = j + 1; k < j + 3 && k < n; k++) {
|
|
||||||
Z[k] += Z[j] * get_content(k, j);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("L-1 * Z\n");
|
|
||||||
// outputZ();
|
|
||||||
|
|
||||||
// calculate Z = D-1 * Z
|
|
||||||
for (int j = 0; j < n; j++) {
|
|
||||||
Z[j] = Z[j] * get_content(j, j);
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("D-1 L-1 * Z\n");
|
|
||||||
// outputZ();
|
|
||||||
|
|
||||||
// calculate Z = U-1 * Z
|
|
||||||
// do this column by column
|
|
||||||
// printf("Starting U-1\n");
|
|
||||||
for (int j = n - 1; j >= 0; j--) {
|
|
||||||
// diag of U is always 1, z[j] * U[j][j] = z[j], so k doesn't have to equal to j
|
|
||||||
for (int k = j - 1; k > j - 3 && k >= 0; k--) {
|
|
||||||
Z[k] += Z[j] * get_content(k, j);
|
|
||||||
}
|
|
||||||
// outputZ();
|
|
||||||
}
|
|
||||||
|
|
||||||
// printf("U-1 D-1 L-1 * Z\n");
|
|
||||||
// outputZ();
|
|
||||||
|
|
||||||
// output
|
|
||||||
for (int j = 0; j < n; j++) {
|
|
||||||
printf("%.4lf ", Z[j]);
|
|
||||||
}
|
|
||||||
printf("\n");
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user