使用列主元消去法进行消元
题号:洛谷3389
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
//A*X=B 已知矩阵A,B 求X
/*
把A消成上三角矩阵,依次回代得到结果
先使第一行化成1 0 0 ……
然后把其他行第一列弄成0 然后把矩阵分块继续弄
*/
const double eps=1E-8;
double X[111][111];
int n;
/*
通常实践表明 选择绝对值较大的元素作为主元 最终误差会较小
数值计算上
*/
void gauss(){
for(int i=1;i<=n;i++){ //枚举未知量
int cur=i;
for(int j=i+1;j<=n;j++)
if(fabs(X[j][i])>fabs(X[cur][i]))
cur=j; //列主元消去法
//找出该列绝对值最大的那个
for(int j=1;j<=n+1;j++)
swap(X[i][j],X[cur][j]);
double tmp=X[i][i];
if(fabs(tmp)<eps) continue;//防0
for(int j=1;j<=n+1;j++)
X[i][j]/=tmp;//化标准形
for(int j=1;j<=n;j++){
if(j==i)
continue;
tmp=X[j][i];
for(int k=1;k<=n+1;k++)
X[j][k]-=tmp*X[i][k];
}
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n+1;j++)
scanf("%lf",&X[i][j]);
gauss();
bool nosol=false,mulsol=false;
for(int i=1;i<=n;i++){
int cnt=0;
for(int j=1;j<=n+1;j++){
if(fabs(X[i][j])<eps) cnt++;//记0
else break;
}
if(cnt==n) nosol=true;
if(cnt==n+1) mulsol=true;
}
if(nosol||mulsol) puts("No Solution");
else{
for(int i=1;i<=n;i++){
printf("%.2lf\n",X[i][n+1]);
}
}
return 0;
}
例题:
[JSOI2008]球形空间产生器sphere
Description
有一个球形空间产生器能够在n维空间中产生一个坚硬的球体。现在,你被困在了这个n维球体中,你只知道球
面上n+1个点的坐标,你需要以最快的速度确定这个n维球体的球心坐标,以便于摧毁这个球形空间产生器。
Input
第一行是一个整数n(1<=N=10)。接下来的n+1行,每行有n个实数,表示球面上一点的n维坐标。每一个实数精确到小数点
后6位,且其绝对值都不超过20000。
Output
有且只有一行,依次给出球心的n维坐标(n个实数),两个实数之间用一个空格隔开。每个实数精确到小数点
后3位。数据保证有解。你的答案必须和标准输出一模一样才能够得分。
Sample Input
2
0.0 0.0
-1.0 1.0
1.0 0.0
Sample Output
0.500 1.500
HINT
提示:给出两个定义:1、 球心:到球面上任意一点距离都相等的点。2、 距离:设两个n为空间上的点A, B
的坐标为(a1, a2, …, an), (b1, b2, …, bn),则AB的距离定义为:dist = sqrt( (a1-b1)^2 + (a2-b2)^2 +
… + (an-bn)^2 )
Source
从二维入手 设球心坐标为(x,y) 给出的三个点分别为(a1,b1),(a2,b2),(a3,b3)
那么
(a1−x)2+(b1−y)2=(a2−x)2+(b2−y)2
(
a
1
−
x
)
2
+
(
b
1
−
y
)
2
=
(
a
2
−
x
)
2
+
(
b
2
−
y
)
2
移项,合并同类项,可以消去
x2和y2
x
2
和
y
2
得
2(a2−a1)+2(b2−b1)=a22+b22−a12−b12
2
(
a
2
−
a
1
)
+
2
(
b
2
−
b
1
)
=
a
2
2
+
b
2
2
−
a
1
2
−
b
1
2
三维就设球心(x,y,z)……以此类推
然后我们就可以根据第一个点的坐标推出其他点的方程
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define Abs(_A) (_A>0?_A:-(_A))
#define eps 1E-8
using namespace std;
int n;
double f[13][13],a[13];
void gauss(){
int i,j,cur,k;
for(i=1;i<=n;i++){
cur=i;
for(j=i+1;j<=n;j++)
if(Abs(f[j][i])>Abs(f[cur][i]))
cur=j;
for(j=1;j<=n+1;j++)
swap(f[i][j],f[cur][j]);//交换绝对值最大行与当前行
double tmp=f[i][i];
if(Abs(tmp)<eps) continue;
for(j=1;j<=n+1;j++)
f[i][j]/=tmp;
for(j=1;j<=n;j++){
if(j==i) continue;
tmp=f[j][i];
for(k=1;k<=n+1;k++)
f[j][k]-=tmp*f[i][k];
}
}
}
int main(){
//保证有解
scanf("%d",&n);
int i,j;double tmp;
for(i=1;i<=n;i++)
scanf("%lf",&a[i]);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++){
scanf("%lf",&tmp);
f[i][j]=2*(tmp-a[j]);
f[i][n+1]+=tmp*tmp-a[j]*a[j];
}
gauss();
for(i=1;i<n;i++)
printf("%.3lf ",f[i][n+1]);
printf("%.3lf",f[n][n+1]);
return 0;
}