列主元高斯消去法.cpp

来自「包含二分法、复化辛卜生公式、改进欧拉法、拉格朗日插值多项式、龙贝格算法、牛顿迭代」· C++ 代码 · 共 50 行

#include<iostream.h>
#include<math.h>

void main()
{
	int i;
	float *x;
	float c[3][4] ={0.101,2.304,3.555,1.183,
					-1.347,3.712,4.623,2.137,
					-2.835,1.072,5.643,3.035};
	float *ColPivot(float *,int);
	x=ColPivot(c[0],3);
	for( i=0;i<=2;i++ )
		cout<<"x("<<i<<")="<<x[i]<<endl;
}

float *ColPivot( float *c,int n )
{
	int i,j,t,k;
	float *x,p;
 	x=new float[n*sizeof(float)];
	for( i=0;i<=n-2;i++)
	{
		k=i;
		for(j=i+1;j<=n-1;j++)
			if(fabs(*(c+j*(n+1)+i))>(fabs(*(c+k*(n+1)+i))))
				k=j;
			if(k!=i)
				for( j=i;j<=n;j++ )
				{
					p=*(c+i*(n+1)+j);
					*(c+i*(n+1)+j)=*(c+k*(n+1)+j);
					*(c+k*(n+1)+j)=p;
				}
		for( j=i+1;j<=n-1;j++ )
		{
			p=(*(c+j*(n+1)+i))/(*(c+i*(n+1)+i));
			for( t=i;t<=n;t++ )
				*(c+j*(n+1)+t)-=p*(*(c+i*(n+1)+t));
		}
	}
	for( i=n-1;i>=0;i--)
	{
		for( j=n-1;j>=i+1;j--)
			(*(c+i*(n+1)+n))-=x[j]*(*(c+i*(n+1)+j));
		x[i]=*(c+i*(n+1)+n)/(*(c+i*(n+1)+i));
	}
	return x;
}