ipzyh

本文出自 “Jeick(编程菜鸟)” 博客，请务必保留此出处http://jeick.blog.51cto.com/713059/146711

#include <cmath>
#include <iostream>
#include <iomanip>
#include <cstdlib>
#include <functional>
#include <vector>
#include <algorithm>
using namespace std;
#define TINY 1.0e-20 //A small number.
void ludcmp(vector<vector<float> > &a, int n, vector<int> &indx, float &d);//对矩阵进行LU分解
void lubksb(vector<vector<float> > &a, int n, vector<int> &indx, vector<float> &b);//对矩阵进行前向和后向回代
void inv(vector<vector<float> > &x,vector<vector<float> > &y);//对矩阵求逆结果保存在y中
void iniv(vector<vector<float> > &x,vector<float> line,int n);//对二维动态数组进行初始化
void main()
{
int i,j,n=4;//输入矩阵的维数
float v[4][4]={{1,0,1,-4},{-1,-3,-4,-2},{2,-1,4,4},{2,3,-3,2}};
vector<vector<float> > x;
vector<float> line;
vector<vector<float> > y;
vector<float> line1;
iniv(y,line1,n);
iniv(x,line,n);
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
x[i].push_back(v[i][j]);
}
}
cout<<"原矩阵:"<<endl;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
cout<<setw(2)<<setiosflags(ios::left)<<setw(2)<<x[i][j]<<" ";
}
cout<<endl;
}
inv(x,y);
cout<<"矩阵的逆:"<<endl;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
cout<<setw(2)<<setiosflags(ios::left)<<setw(2)<<y[i][j]<<" ";
}
cout<<endl;
}
}
void inv(vector<vector<float> > &x,vector<vector<float> > &y)
{
int n=x.size(),i=0,j=0;
vector<int> index(n);
index.clear();
float m=1.0;
ludcmp(x,n,index,m);
vector<float> col(n);
for(j=0;j<n;j++)
{
for(i=0;i<n;i++)
col[i]=0.0;
col[j]=1.0;
lubksb(x,n,index,col);
for(i=0;i<n;i++)
{
y[i].push_back(col[i]);
}