lu.cpp

Lu 方法解线性方程组的程序 可以参考一下

#include<stdio.h>
#include<iostream.h>
#include<fstream.h>
void main()
{
	ifstream fin("input.txt",ios::in);
	ofstream fout("output.txt",ios::out);
	double **a,**c,*x;
	int n,i,j,k,flag=1;
	fin>>n;
	a=new double*[n];
	c=new double*[n];
	x=new double[n];
	for(i=0;i<n;++i)
	{
		a[i]=new double[n];
		c[i]=new double[n];
	}
	for(i=0;i<n;++i)
	{
		for(j=0;j<n;++j)
		{
			fin>>a[i][j];
			c[i][j]=a[i][j];
		}
		fin>>x[i];
	}

	if(c[0][0]==0)
	{
		fout<<"LU分解法无法运行！"<<endl;
		flag=0;
	}
	if(flag==1)
	{
		for(i=1;i<n;++i)
			c[i][0]=a[i][0]/c[0][0];
		for(i=1;i<n;++i)
		{
			for(j=i;j<n;++j)
			{
				for(k=0;k<i;++k)
					c[i][j]=c[i][j]-c[i][k]*c[k][j];
			}
			if(c[i][i]==0)
			{
				fout<<"LU分解法无法运行！"<<endl;
				flag=0;
				break;
			}
			if(flag==1)
			{
				for(j=i+1;j<n;++j)
				{
					for(k=0;k<i;++k)
						c[j][i]=c[j][i]-c[j][k]*c[k][i];
					c[j][i]=c[j][i]/c[i][i];
				}
			}
		}
	}
	if(flag==1)
	{
		for(i=1;i<n;++i)
			for(j=0;j<i;++j)
				x[i]=x[i]-c[i][j]*x[j];
		x[n-1]=x[n-1]/c[n-1][n-1]; 
		for(i=n-2;i>=0;--i)
		{
			for(j=i+1;j<n;++j)
				x[i]=x[i]-c[i][j]*x[j];
			x[i]=x[i]/c[i][i];
		}
		for(i=0;i<n;++i)
			fout<<x[i]<<endl;
	}
	delete[] a;
	delete[] c;
	delete[] x;
	fin.close();
	fout.close();
}