问题描述
为求解一个线性方程组,首先构造增广矩阵[A|B],采用偏序选主元策略的高斯消去法变换成上三角矩阵,再执行回代过程得到解。
【输入形式】在屏幕上依次输入方阵阶数n,系数矩阵A和常数矩阵B。
【输出形式】首先输出上三角矩阵(变换后的增广矩阵),然后每一行输出一个根
【样例1输入】
4
1 2 1 4
2 0 4 3
4 2 2 1
-3 1 3 2
13
28
20
6
【样例1输出】
4 2 2 1 20
0 2.5 4.5 2.75 21
0 0 4.8 3.6 26.4
0 0 0 3.75 7.5
3
-1
4
2
【样例1说明】输入:第1行为方阵阶数4,第2行至5行为系数矩阵A,第6行至9行为常数矩阵B。
输出:首先输出上三角矩阵(变换后的增广矩阵),然后每行依次输出方程解:x1, x2, x3, x4。
【样例2输入】
4
1 2 1 4
2 4 2 8
4 2 2 1
-3 1 3 2
13
28
20
6
【样例2输出】
error
解题思路
重点在于理解如何化三角矩阵。原理与之类似,只不过求出其三角矩阵即可。
c++源代码
#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;
int a[10][10]; //系数矩阵A
int b[10]; //常数矩阵B
double c[10][10]; //增广矩阵C
double x[10] = { 0.0 };//跟数组
//高斯消去
int gauss(int n)
{
int el, ef = 0;
for (int i = 0; i < n - 1; i++)
{
//交换每一列的最大值
double x = fabs(c[i][i]);
for (int m = i; m < n ; m++)
{
if (fabs(c[m][i]) > x)
{
x = fabs(c[m][i]);
ef = 1;
el = m;
}
}
if (ef)
{
for (int j = i; j <= n; j++)
swap(c[i][j], c[el][j]);
}
//化三角矩阵
for (int k = i+1; k < n; k++)
{
double s;
s = -c[k][i] / c[i][i] * 1.0;
for (int j = i; j <= n; j++)
c[k][j] = c[k][j] + s * c[i][j];
}
}
//判断是否为奇异矩阵
for (int i = 0; i < n; i++)
{
if (c[i][i] == 0)
{
cout << "error";
return 0;
}
}
//输出三角矩阵
for (int i = 0; i < n; i++)
{
for (int j = 0; j <= n; j++)
cout << c[i][j] << " ";
cout << endl;
}
cout << endl;
return 1;
}
//迭代求值
void juzhen(int n)
{
for (int i = n -1; i >= 0; i--)
{
double tmp = c[i][n];
for (int j = 0; j < n; j++)
tmp -= c[i][j] * x[j];
x[i] = 1.0 * tmp / c[i][i];
}
for (int i = 0; i < n; i++)
{
cout << x[i] << endl;
}
}
int main()
{
int n;
cin >> n;
cout << endl;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
cin >> a[i][j];
cout << endl;
}
cout << endl;
for (int i = 0; i < n; i++)
{
cin >> b[i];
cout << endl;
}
cout << endl;
//将矩阵A、B合并为增广矩阵C
for (int i = 0; i < n; i++)
{
for (int j=0 ; j < n; j++)
c[i][j] = a[i][j];
c[i][n] = b[i];
cout << endl;
}
int q=gauss(n);
if(q)
juzhen(n);
return 0;
}
运行结果: