P4035 [JSOI2008]球形空间产生器 ——高斯消元

Strezia

已于 2022-02-21 15:56:41 修改

阅读量194

点赞数 1

分类专栏：洛谷数学文章标签：矩阵线性代数 c++

于 2022-02-12 23:15:58 首次发布

本文链接：https://blog.csdn.net/m0_59273843/article/details/122903736

版权

数学同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

洛谷

22 篇文章 0 订阅

订阅专栏

Link
一道高斯消元的题目，其实重点还是在如何得出方程组也就是增广矩阵。
在这里插入图片描述

代码

const double EPS=1e-7;
inline int gauss(double a[][N], bool l[], double ans[], const int& n) {
    int res = 0, r = 0;
    for(int i = 0; i < n; i++)
        l[i] = false;
    for(int i = 0; i < n; i++) {
        for(int j = r; j < n; j++)
            if(fabs(a[j][i]) > EPS) {
                for(int k = i; k <= n; k++)
                    swap(a[j][k], a[r][k]);
                break;
            }
        if(fabs(a[r][i]) < EPS) {
            ++res;
            continue;
        }
        for(int j = 0; j < n; j++)
            if(j != r && fabs(a[j][i]) > EPS) {
                double tmp = a[j][i] / a[r][i];
                for(int k = i; k <= n; k++)
                    a[j][k] -= tmp * a[r][k];
            }
        l[i] = true;
        ++r;
    }
    for(int i = 0; i < n; i++)
        if(l[i])
            for(int j = 0; j < n; j++)
                if(fabs(a[j][i]) > EPS)
                    ans[i] = a[j][n] / a[j][i];
    return res;
}
double b[N][N], a[N][N];
double ans[N];
bool l[N];
int n;
void solve() {
    cin >> n;
    for(int i = 0; i < n + 1; i++) {
        for(int j = 0; j < n; j++) {
            cin >> b[i][j];
        }
    }
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++)
            a[i][n] += b[i][j] * b[i][j] - b[i+1][j]*b[i+1][j];
    }
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            a[i][j] = 2 * (b[i][j] - b[i+1][j]);
        }
    }
    gauss(a, l, ans, n);
    for(int i = 0; i < n; i++)
        printf("%.3lf ", ans[i]);
    cout << endl;
}