#include "svd.h"
/*********************************************************************
* 矩阵的奇异值分解,参见《c 常用算法程序集》徐世良P169
* 参数说明:
* a m*n的实矩阵,返回时其对角线给出奇异值(非递增顺序),其余元素为0
* m,n 矩阵A的行数和列数
* u m*m的矩阵,存放左奇异向量
* v n*n的矩阵,存放右奇异向量
* eps 双精度实型变量,给定精度要求
* ka 整形变量,其值为max(m,n)+1
* 返回值:如果返回标志小于0,则说明出现了迭代MAX_ITERA次还未求得某个
* 奇异值的情况,此时矩阵A的分解式为UAV,如果返回标志大于0,则说明
* 程序正常运行
********************************************************************/
int dluav(double a[],int m,int n,double u[],double v[],double eps,int ka)
{
int i,j,k,l,it,ll,kk,ix,iy,mm,nn,iz,ml,ks;
double d,dd,t,sm,sml,eml,sk,ek,b,c,shh,fg[2],cs[2];
double *s,*e,*w;
s=(double*)malloc(ka*sizeof(double));
e=(double*)malloc(ka*sizeof(double));
w=(double*)malloc(ka*sizeof(double));
for(i=1;i<=m;i++)
{
ix=(i-1)*m+i-1;
u[ix]=0;
}
for(i=1;i<=n;i++)
{
iy=(i-1)*n+i-1;
v[iy]=0;
}
it=MAX_ITERA;k=n;
if(m-1<n)
k=m-1;
l=m;
if(n-2<m) l=n-2;
if(l<0) l=0;
ll=k;
if(l>k) ll=l;
if(ll>=1)
{
for(kk=1;kk<=ll;kk++)
{
if(kk<=k)
{
d=0.0;
for(i=kk;i<=m;i++)
{
ix=(i-1)*n+kk-1;d=d+a[ix]*a[ix];
}
s[kk-1]=sqrt(d);
//if(s[kk-1]!=0.0)
if(fabs(s[kk-1])>MIN_DOUBLE)
{
ix=(kk-1)*n+kk-1;
//if(a[ix]!=0.0)
if(fabs(a[ix])>MIN_DOUBLE)
{
s[kk-1]=fabs(s[kk-1]);
if(a[ix]<0.0) s[kk-1]=-s[kk-1];
}
for(i=kk;i<=m;i++)
{
iy=(i-1)*n+kk-1;
a[iy]=a[iy]/s[kk-1];
}
a[ix]=1.0+a[ix];
}
s[kk-1]=-s[kk-1];
}
if(n>=kk+1)
{
for(j=kk+1;j<=n;j++)
{
//if((kk<=k)&&(s[kk-1]!=0.0))
if((kk<=k)&&(fabs(s[kk-1])>MIN_DOUBLE))
{
d=0.0;
for(i=kk;i<=m;i++)
{
ix=(i-1)*n+kk-1;
iy=(i-1)*n+j-1;
d=d+a[ix]*a[iy];
}
d=-d/a[(kk-1)*n+kk-1];
for(i=kk;i<=m;i++)
{
ix=(i-1)*n+j-1;
iy=(i-1)*n+kk-1;
a[ix]=a[ix]+d*a[iy];
}
}
e[j-1]=a[(kk-1)*n+j-1];
}
}
if(kk<=k)
{
for(i=kk;i<=m;i++)
{
ix=(i-1)*m+kk-1;iy=(i-1)*n+kk-1;
u[ix]=a[iy];
}
}
if(kk<=l)
{
d=0.0;
for(i=kk+1;i<=n;i++)
d=d+e[i-1]*e[i-1];
e[kk-1]=sqrt(d);
//if(e[kk-1]!=0.0)
if(fabs(e[kk-1])>MIN_DOUBLE)
{
//if(e[kk]!=0.0)
if(fabs(e[kk])>MIN_DOUBLE)
{
e[kk-1]=fabs(e[kk-1]);
if(e[kk]<0.0)
e[kk-1]=-e[kk-1];
}
for(i=kk+1;i<=n;i++)
e[i-1]=e[i-1]/e[kk-1];
e[kk]=1.0+e[kk];
}
e[kk-1]=-e[kk-1];
//if((kk+1<=m)&&(e[kk-1]!=0.0))
if((kk+1<=m)&&(fabs(e[kk-1])>MIN_DOUBLE))
{
for(i=kk+1;i<=m;i++) w[i-1]=0.0;
for(j=kk+1;j<=n;j++)
for(i=kk+1;i<=m;i++)
w[i-1]=w[i-1]+e[j-1]*a[(i-1)*n+j-1];
for(j=kk+1;j<=n;j++)
for(i=kk+1;i<=m;i++)
{
ix=(i-1)*n+j-1;
a[ix]=a[ix]-w[i-1]*e[j-1]/e[kk];
}
}
for(i=kk+1;i<=n;i++)
v[(i-1)*n+kk-1]=e[i-1];
}
}
}
mm=n;
if(m+1<n) mm=m+1;
if(k<n) s[k]=a[k*n+k];
if(m<mm) s[mm-1]=0.0;
if(l+1<mm) e[l]=a[l*n+mm-1];
e[mm-1]=0.0;
nn=m;
if(m>n) nn=n;
if(nn>=k+1)
{
for(j=k+1;j<=nn;j++)
{
for(i=1;i<=m;i++)
u[(i-1)*m+j-1]=0.0;
u[(j-1)*m+j-1]=1.0;
}
}
if(k>=1)/////////////////////////////////
{
for(ll=1;ll<=k;ll++)
{
kk=k-ll+1;iz=(kk-1)*m+kk-1;
//if(s[kk-1]!=0.0)
if(fabs(s[kk-1])>MIN_DOUBLE)
{
if(nn>=kk+1)
for(j=kk+1;j<=nn;j++)
{
d=0.0;
for(i=kk;i<=m;i++)
{
ix=(i-1)*m+kk-1;
iy=(i-1)*m+j-1;
d=d+u[ix]*u[iy]/u[iz];
}
d=-d;
for(i=kk;i<=m;i++)
{
ix=(i-1)*m+j-1;
iy=(i-1)*m+kk-1;
u[ix]=u[ix]+d*u[iy];
}
}
for(i=kk;i<=m;i++)
{
ix=(i-1)*m+kk-1;
u[ix]=-u[ix];
}
u[iz]=1.0+u[iz];
if(kk-1>=1)//////////////////////////////////////
for(i=1;i<=kk-1;i++)
u[(i-1)*m+kk-1]=0.0;
}
else
{
for(i=1;i<=m;i++)
u[(i-1)*m+kk-1]=0.0;
u[(kk-1)*m+kk-1]=1.0;
}
}
}
for(ll=1;ll<=n;ll++)
{
kk=n-ll+1;iz=kk*n+kk-1;
//if((kk<=l)&&(e[kk-1]!=0.0))/////////////////////////////
if((kk<=l)&&(fabs(e[kk-1])>MIN_DOUBLE))
{
for(j=kk+1;j<=n;j++)
{
d=0.0;
for(i=kk+1;i<=n;i++)
{
ix=(i-1)*n+kk-1;iy=(i-1)*n+j-1;
d=d+v[ix]*v[iy]/v[iz];
}
d=-d;
for(i=kk+1;i<=n;i++)
{
ix=(i-1)*n+j-1;iy=(i-1)*n+kk-1;
v[ix]=v[ix]+d*v[iy];
}
}
}
for(i=1;i<=n;i++)
v[(i-1)*n+kk-1]=0.0;
v[iz-n]=1.0;
}
for(i=1;i<=m;i++)
for(j=1;j<=n;j++)
a[(i-1)*n+j-1]=0.0;
ml=mm;
it=MAX_ITERA;
while(1==1)//////////////////////////////////
{
if(mm==0)
{
ppp(a,e,s,v,m,n);
free(s);free(e);free(w);
return l;
}
if(it==0)
{
ppp(a,e,s,v,m,n);
free(s);free(e);free(w);
return -1;
}
kk=mm-1;
//while((kk!=0)&&(fabs(e[kk-1])!=0.0))
while((kk!=0)&&(fabs(e[kk-1])>MIN_DOUBLE))
{
d=fabs(s[kk-1])+fabs(s[kk]);
dd=fabs(e[kk-1]);
if(dd>eps*d)
kk=kk-1;
else
e[kk-1]=0.0;
}
if(kk==mm-1)
{
kk=kk+1;
if(s[kk-1]<0.0)
{
s[kk-1]=-s[kk-1];
for(i=1;i<=n;i++)
{
ix=(i-1)*n+kk-1;
v[ix]=-v[ix];
}
}
while((kk!=ml)&&(s[kk-1]<s[kk]))
{
d=s[kk-1];s[kk-1]=s[kk];s[kk]=d;
if(kk<n)
for(i=1;i<=n;i++)
{
ix=(i-1)*n+kk-1;iy=(i-1)*n+kk;
d=v[ix];v[ix]=v[iy];v[iy]=d;
}
if(kk<m)
for(i=1;i<=m;i++)
{
ix=(i-1)*m+kk-1;
iy=(i-1)*m+kk;
d=u[ix];u[ix]=u[iy];u[iy]=d;
}
kk=kk+1;
}
it=MAX_ITERA;
mm=mm-1;
}
else
{
ks=mm;
//while((ks>kk)&&(fabs(s[ks-1])!=0.0))
while((ks>kk)&&(fabs(s[ks-1])>MIN_DOUBLE))
{
d=0.0;
if(ks!=mm)
d=d+fabs(e[ks-1]);
if(ks!=kk+1) d=d+fabs(e[ks-2]);
dd=fabs(s[ks-1]);
if(dd>eps*d)
ks=ks-1;
else
s[ks-1]=0.0;
}
if(ks==kk)
{
kk=kk+1;
d=fabs(s[mm-1]);
t=fabs(s[mm-2]);
if(t>d)
d=t;
t=fabs(e[mm-2]);
if(t>d)
d=t;
t=fabs(s[kk-1]);
if(t>d)
d=t;
t=fabs(e[kk-1]);
if(t>d)
d=t;
sm=s[mm-1]/d;sml=s[mm-2]/d;
eml=e[mm-2]/d;
sk=s[kk-1]/d;ek=e[kk-1]/d;
b=((sml+sm)*(sml-sm)+eml*eml)/2.0;
c=sm*eml;c=c*c;shh=0.0;
//if((b!=0.0)||(c!=0.0))
if((fabs(b)>MIN_DOUBLE)||(fabs(c)>MIN_DOUBLE))
{
shh=sqrt(b*b+c);
if(b<0.0)
shh=-shh;
shh=c/(b+shh);
}
fg[0]=(sk+sm)*(sk-sm)-shh;
fg[1]=sk*ek;
for(i=kk;i<=mm-1;i++)
{
sss(fg,cs);
if(i!=kk)
e[i-2]=fg[0];
fg[0]=cs[0]*s[i-1]+cs[1]*e[i-1];
e[i-1]=cs[0]*e[i-1]-cs[1]*s[i-1];
fg[1]=cs[1]*s[i];
s[i]=cs[0]*s[i];
//if((cs[0]!=1.0)||(cs[1]!=0.0))
if((fabs(cs[0]-1.0)>MIN_DOUBLE)||(fabs(cs[1])>MIN_DOUBLE))
for(j=1;j<=n;j++)
{
ix=(j-1)*n+i-1;
iy=(j-1)*n+i;
d=cs[0]*v[ix]+cs[1]*v[iy];
v[iy]=-cs[1]*v[ix]+cs[0]*v[iy];
v[ix]=d;
}
sss(fg,cs);
s[i-1]=fg[0];
fg[0]=cs[0]*e[i-1]+cs[1]*s[i];
s[i]=-cs[1]*e[i-1]+cs[0]*s[i];
fg[1]=cs[1]*e[i];
e[i]=cs[0]*e[i];
if(i<m)
//if((cs[0]!=1.0)||(cs[1]!=0.0))
if((fabs(cs[0]-1.0)>MIN_DOUBLE)||(fabs(cs[1])>MIN_DOUBLE))
for(j=1;j<=m;j++)
{
ix=(j-1)*m+i-1;
iy=(j-1)*m+i;
d=cs[0]*u[ix]+cs[1]*u[iy];
u[iy]=-cs[1]*u[ix]+cs[0]*u[iy];
u[ix]=d;
}
}
e[mm-2]=fg[0];
it=it-1;
}
else
{
if(ks==mm)
{
kk=kk+1;
fg[1]=e[mm-2];e[mm-2]=0.0;
for(ll=kk;ll<=mm-1;ll++)
{
- 1
- 2
- 3
- 4
- 5
- 6
前往页