QR.zip_qr

#include<iostream.h> #include<math.h> #define n 3 #define e 1e-6 int hold_r(float a[n][n],int r) { for(int i=r+1;i<n;i++) { if (a[i][r]!=0) return 1; } } float caculate_cr(float a[n][n],int r) { float ans=0; for(int i=r;i<n;i++) { ans+=(a[i][r]*a[i][r]); } // cout<<"ans"<<ans<<endl; if (a[r][r]<=0) return sqrt(ans); else return -sqrt(ans); } float caculate_hr(float a[n][n],int r,float cr) { return cr*(cr-a[r][r]); } void caculate_ur(float a[n][n],int r,float u[n],float cr) { for (int j=0;j<r;j++) /// return 0 { u[j]=0; } u[r]=(a[r][r]-cr); for(int i=r+1;i<n;i++) { u[i]=a[i][r]; } // cout<<"Ur"<<r<<"\t"; // for(int k=0;k<n;k++) // { // cout<<u[k]<<"\t"; // } // cout<<endl; } void caculate_vr(float vr[n],float ur[n],int r,float hr) { for (int j=0;j<r;j++) /// return 0 { vr[j]=0; } for(int i=r;i<n;i++) { vr[i]=ur[i]/hr; } // cout<<"vr"<<r<<"\t"; // for(int k=0;k<n;k++) // { // cout<<vr[k]<<"\t"; // } // cout<<endl; } void multitude_ax(float a[n][n],int r,float x[n],float v[n]) { for (int i=0;i<n;i++) { x[i]=0; } for (i=0;i<n;i++) { for (int j=r;j<n;j++) { x[i]+=a[i][j]*v[j]; } } } void turn_a(float a[n][n]) { float t=0; for (int i=0;i<n;i++) { for (int j=i+1;j<n;j++) { t=a[i][j]; a[i][j]=a[j][i]; a[j][i]=t; } } } //float caculate_tr(float vr[n],float pr[n]) //{ /// float ans=0; // for(int i=0;i<n;i++) // { // ans+=vr[i]*pr[i]; // } // return ans; //} void multitude_xxt(float a[n][n],float x1[n],float x2[n]) //x1[n]*x2 T==pa; a[n][n]-pa[n][n] { float temp=0; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { temp=x1[i]*x2[j]; a[i][j]=a[i][j]-temp; } } } //void caculate_wr(float pr[n],float ur[n],float tr) //{ // for(int i=0;i<n;i++) // { // pr[i]-=(tr*ur[i]); // } //} void check_small_aij(float a[n][n]) { double t=0; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { double t=a[i][j]; if(fabs(t)<e) { a[i][j]=0; } else ; // cout<<a[i][j]<<'\t'; } // cout<<endl; } } void show(float a[n][n]) { for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { cout<<a[i][j]<<'\t'; } cout<<endl; } } void metrix_r(float a[n][n],float ur[n],float vr[n],float pr[n]) { for (int r=0;r<n-1;r++) { if(hold_r(a,r)!=0) //// (1) { float cr=caculate_cr(a,r); ///(2) float hr=caculate_hr(a,r,cr); caculate_ur(a,r,ur,cr); caculate_vr(vr,ur,r,hr); turn_a(a); multitude_ax(a,r,pr,vr); ///caculate pr[n] turn_a(a); multitude_xxt(a,ur,pr); // check_small_aij(a); // cout<<r<<endl; // show(a); } else ; } } void main() { float A[n][n]={{1,2,2},{2,1,2},{1,2,1}}; float ur[n]={0,0,0}; float vr[n]={0,0,0}; float pr[n]={0,0,0}; metrix_r(A,ur,vr,pr); check_small_aij(A); show(A); }