6.4 求解复系数方程组的全选主元高斯-约当消去法 cjdn.c

许士良常用算法程序集C语言,包括c++一些常用算法代码

#include "stdlib.h"
#include "stdio.h"

int cjdn(ar,ai,br,bi,n,m)
int n,m;
double ar[],ai[],br[],bi[];
{ 
	int *js,l,k,i,j,is,u,v;
	double p,q,s,d;
	js=malloc(n*sizeof(int));
	for(k=0;k<=n-1;k++)
	{ 
		d=0.0;
		for(i=k;i<=n-1;i++)
			for(j=k;j<=n-1;j++)
			{ 
				u=i*n+j;
				p=ar[u]*ar[u]+ai[u]*ai[u];
				if (p>d) 
				{
					d=p;
					js[k]=j;
					is=i;
				}
			 }
		if (d+1.0==1.0)
		{ 
			free(js); 
			printf("err**fail\n");
			return(0);
		}
		if(is!=k)
		{ 
			for(j=k;j<=n-1;j++)
			{ 
				u=k*n+j; v=is*n+j;
				p=ar[u]; ar[u]=ar[v]; ar[v]=p;
				p=ai[u]; ai[u]=ai[v]; ai[v]=p;
			}
			for(j=0;j<=m-1;j++)
			{ 
				u=k*m+j; v=is*m+j;
				p=br[u]; br[u]=br[v]; br[v]=p;
				p=bi[u]; bi[u]=bi[v]; bi[v]=p;
			}
		}
		if(js[k]!=k)
		for(i=0;i<=n-1;i++)
        { 
			u=i*n+k; v=i*n+js[k];
			p=ar[u]; ar[u]=ar[v]; ar[v]=p;
			p=ai[u]; ai[u]=ai[v]; ai[v]=p;
        }
		v=k*n+k;
		for (j=k+1;j<=n-1;j++)
		{ 
			u=k*n+j;
			p=ar[u]*ar[v]; q=-ai[u]*ai[v];
			s=(ar[v]-ai[v])*(ar[u]+ai[u]);
			ar[u]=(p-q)/d; ai[u]=(s-p-q)/d;
		}
		for (j=0;j<=m-1;j++)
		{ 
			u=k*m+j;
			p=br[u]*ar[v]; q=-bi[u]*ai[v];
			s=(ar[v]-ai[v])*(br[u]+bi[u]);
			br[u]=(p-q)/d; bi[u]=(s-p-q)/d;
		}
		for (i=0;i<=n-1;i++)
		{
			if (i!=k)
			{ 
				u=i*n+k;
				for (j=k+1;j<=n-1;j++)
				{ 
					v=k*n+j; l=i*n+j;
					p=ar[u]*ar[v]; q=ai[u]*ai[v];
					s=(ar[u]+ai[u])*(ar[v]+ai[v]);
					ar[l]=ar[l]-p+q;
					ai[l]=ai[l]-s+p+q;
				}
				for (j=0;j<=m-1;j++)
				{ 
					l=i*m+j; v=k*m+j;
					p=ar[u]*br[v]; q=ai[u]*bi[v];
					s=(ar[u]+ai[u])*(br[v]+bi[v]);
					br[l]=br[l]-p+q; bi[l]=bi[l]-s+p+q;
				}
			}
		}
	}
	for(k=n-1;k>=0;k--)
	{
		if(js[k]!=k)
		{
			for(j=0;j<=m-1;j++)
			{ 
				u=k*m+j;v=js[k]*m+j;
				p=br[u]; br[u]=br[v]; br[v]=p;
				p=bi[u]; bi[u]=bi[v]; bi[v]=p;
			}
		}
	}
	free(js);
	return(1);
}