部分透视/高斯消除-交换列而不是产生错误输出的行

时间:2019-01-31 06:04:07

标签: c++

我正在尝试实施一个快速程序来求解线性方程组。该程序从文件读取输入,然后将上三角系统和解决方案写入文件。它没有任何透视,但是当我尝试实现透视时,会产生错误的结果。 作为示例输入,这是以下方程组:

w+2x-3y+4z=12
2w+2x-2y+3z=10
x+y=-1
w-x+y-2z=-4

我希望结果为w = 1,x = 0,y = -1和z = 2。当我不旋转时,会得到这个答案(x上有一些舍入错误)。当我加上透视时,我得到相同的数字,但顺序错误:w = 2,x = 1,y = -1和z = 0。

我需要怎么做才能以正确的顺序获得这些?我在某处错过了一步吗?我需要做列交换而不是行交换,因为我需要在以后需要它的情况下使它适应于并行算法。这是执行消除和反向替换的代码:

void gaussian_elimination(double** A, double* b, double* x, int n)
{
    int maxIndex;
    double temp;
    int i;
    for (int k = 0; k < n; k++)
    {               
        i = k;
        for (int j = k+1; j < n; j++)
        {
            if (abs(A[k][j]) > abs(A[k][i]))
            {
                i = j;
            }
        }       

        if (i != k)
        {
            for (int j = 0; j < n; j++)
            {
                temp = A[j][k];
                A[j][k] = A[j][i];
                A[j][i] = temp;
            }           
        }


        for (int j = k + 1; j < n; j++)
        {
            A[k][j] = A[k][j] / A[k][k];
        }
        b[k] = b[k] / A[k][k];
        A[k][k] = 1;
        for (i = k + 1; i < n; i++)
        {
            for (int j = k + 1; j < n; j++)
            {
                A[i][j] = A[i][j] - A[i][k] * A[k][j];
            }
            b[i] = b[i] - A[i][k] * b[k];
            A[i][k] = 0;
        }
    }

}

void back_substitution(double**U, double*x, double*y, int n)
{
    for (int k = n - 1; k >= 0; k--)
    {
        x[k] = y[k];
        for (int i = k - 1; i >= 0; i--)
        {
            y[i] = y[i] - x[k]*U[i][k];
        }
    }
}

1 个答案:

答案 0 :(得分:0)

我相信您实施的实际上是完整的数据透视。

要完全旋转,必须跟踪列的排列,并将相同的排列应用于答案。

您可以使用数组{0,1,...,n},在第二个循环中交换第i和第k个值。然后,使用此数组重新排列解决方案。

如果您要执行的是部分透视,则需要在相应的行中查找最大值,并相应地交换行和'b'的值。