高斯消元模板 TZOJ 5817: 球形空间产生器

关于高斯消元：

时间复杂度O(n^3)
利用增广矩阵可以求出线性方程。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 105;
const double eps = 1e-6;
double a[N][N];
int n;
int gauss()
{
    int r,c;
    for(r=1,c=1;c<=n;c++)
    {
        int t=r;
        for(int i=r;i<=n;i++)
        {
            if(fabs(a[i][c])>fabs(a[t][c]))
            {
                t=i;
            }
        }
        if(fabs(a[t][c])<eps) continue;
        for(int i=c;i<=n+1;i++)swap(a[r][i],a[t][i]);
        for(int i=n+1;i>=c;i--)a[r][i]/=a[r][c];
        for(int i=r+1;i<=n;i++)
        {
            if(fabs(a[i][c])>eps)
            {
                for(int j=n+1;j>=c;j--)a[i][j]-=a[i][c]*a[r][j];
            }
        }
        r++;
    }
    if(r<=n)
    {
        for(int i=r;i<=n;i++)
        {
            if(fabs(a[i][n+1])>eps)return 2;
        }
        return 1;
    }
    for(int i=n;i>=1;i--)
    {
        for(int j=n;j>=i+1;j--)
        {
            a[i][n+1]-=a[j][n+1]*a[i][j];
        }
    }
    return 0;
}
int main()
{
    scanf("%d", &n);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n+1;j++)
        {
            scanf("%lf", &a[i][j]);
        }
    }
    int t=gauss();
    if (t == 0)
    {
        for (int i = 1; i <=n; i ++ ) printf("%.2lf\n", a[i][n+1]);
    }
    else if (t == 1) puts("Infinite group solutions");
    else puts("No solution");
}

题目：

题解：

利用n+1个式子求出一个线性方程组。答案保证有解。

代码

#include<bits/stdc++.h>
using namespace std;
double a[15][15],b[15][15]; 
int n;

void gauss()
{
    for(int r=1,c=1;c<=n;c++,r++)
    {
        int t=r;
        for(int i=r+1;i<=n;i++)
        {
            if(fabs(b[i][c])>fabs(b[t][c]))
            {
                t=i;
            }
        }
        for(int i=c;i<=n+1;i++)swap(b[t][i],b[r][i]);
        for(int i=n+1;i>=c;i--)b[r][i]/=b[r][c];
        for(int i=r+1;i<=n;i++)
        {
            for(int j=n+1;j>=c;j--)
            {
                b[i][j]-=b[i][c]*b[r][j];
            }
        }
    }
    for(int i=n;i>1;i--)
    {
        for(int j=i-1;j;j--)
        {
            b[j][n+1]-=b[i][n+1]*b[j][i];
            b[j][i]=0;
        }
    }
}

int main()
{
   
    scanf("%d", &n);
    for(int i=0;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%lf",&a[i][j]);
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            b[i][j]=2*(a[0][j]-a[i][j]);
            b[i][n+1]+=a[0][j]*a[0][j]-a[i][j]*a[i][j];
        }
    }
    gauss();
    
    for(int i=1;i<n;i++)printf("%.3lf ",b[i][n+1]);
    printf("%.3lf\n",b[n][n+1]);
}