高斯消去法

代码

#include "stdio.h"
#include "math.h"
#include "stdlib.h"
typedef double arrtype;
//高斯消去法
void GuassScale(arrtype coff[],int n,arrtype *x = NULL,int m = 1)
{
int col = n + m;
arrtype *s = new arrtype[n];
int *nrow = new int[n];
arrtype *base = coff;
for(int i = 0; i<n;i++)
{
  s[i] = 0;
  for(int j = 0;j<n;j++)
  {
   arrtype t = fabs(coff[i*col + j]);
   if(s[i] < t)
    s[i] = t;
  }
  nrow[i] = i;
}
for(i=0;i<n-1;i++)
{
  int p = i;
  arrtype t = fabs(coff[nrow[p]*col+i])/s[nrow[p]];
  for(int j= i+1;j<n;j++)
  {
   arrtype temp= fabs(coff[nrow[j]*col+i])/s[nrow[j]];
   if(t<temp)
   {
    t=temp;
    p=j;
   }    
  }
  if(coff[nrow[p]*col+i]==0)
   return;
  if(nrow[p]!=nrow[i])
  {
   int ncopy = nrow[i];
   nrow[i] = nrow[p];
   nrow[p] =ncopy;
  }
  for(j= i+1;j<n;j++)
  {
   arrtype m = coff[nrow[j]*col+i]/coff[nrow[i]*col+i];
   for(int k=i+1;k<col;k++)
   {
    coff[nrow[j]*col+k]-=coff[nrow[i]*col+k]*m;
   }
  }
}
if(coff[nrow[n-1]*col+n-1]==0)
  return;
for(i=n;i<col;i++)
{
  x[(n-1)*m+(i-n)]=coff[nrow[n-1]*col+i]/coff[nrow[n-1]*col+n-1];
}
for(i=n-2;i>=0;i--)
{
  for(int j = n;j<col;j++)
  {
   for(int k=n-1;k>i;k--)
    coff[nrow[i]*col+j]-=coff[nrow[i]*col+k]*x[k*m+(j-n)];
   x[i*m+(j-n)]=coff[nrow[i]*col+j]/coff[nrow[i]*col+i];
  }
}
}
void main(int argc,char* argv[])
{
FILE *fpi,*fpo;
if(argc > 1)
  fpi=fopen(argv[1],"r");
else
  fpi=stdin;
if(!fpi)return;
if(argc>2)
  fpo = fopen(argv[2],"w");
else
  fpo = stdout;
if(!fpo)return;
int m,n;
printf("请输入ｎ和ｍ的值：\n");
fscanf(fpi,"%d",&n);
fscanf(fpi,"%d",&m);
arrtype *co;
co = new arrtype[n*(n+m)];
for(int i = 0;i<n*(n+m);i++)
{
  float t;
  fscanf(fpi,"%f",&t);
  co[i]=t;
}
fclose(fpi);
arrtype *x = new arrtype[m*n];
GuassScale(co,n,x,m);
delete[] co;
for(i=0;i<n;i++)
{
  fprintf(fpo,"x[%d]=",i);
  for(int j=0;j<m;j++)
   fprintf(fpo,"%f",x[i*m+j]);
  fprintf(fpo,"\n");
}
delete[] x;
fclose(fpo);
} 

 

转载于:https://www.cnblogs.com/magic_evan/archive/2010/10/11/1847527.html