高斯消元(浮点)

//接口:Gauss();解存在x数组中。equ,var分别表示方程个数和变量的个数,(因为是变量 的个数不算常数的个数)要赋值。a就是行列式,其中常数是在方程右侧的符号。
int const maxn = 2000;
const double eps=1e-10;
double a[maxn][maxn],x[maxn];
int equ,var;
int Gauss()
{
    int i, j, k, max_r, col;
    double tmp;
    col = 0;

    for(k = 0; k<equ && col<var; k++, col++)
    {
        max_r = k;
        for(i = k+1; i < equ; i++)
            if(fabs(a[i][col])-fabs(a[max_r][col]) > eps)
            max_r = i;

        if(max_r != k)
            for(j = k; j < var+1; j++)
            swap(a[k][j], a[max_r][j]);

        if(fabs(a[k][col]) < eps)
        {
            k--;
            continue;
        }
        for(i = k+1; i < equ; i++)
        {
            if(fabs(a[i][col]) > eps)
            {
                double t = a[i][col]/a[k][col];  //这里和整型的不同。
                a[i][col] = 0.0;

                for(j = col; j < var+1; j++)
                a[i][j] -= a[k][j]*t;
            }
        }
    }
    for(i = var-1; i >= 0; i--)
    {
        if(fabs(a[i][i]) < eps) continue;
        tmp = a[i][var];
        for(j = i+1; j < var; j++)
        if(a[i][j] != 0)
        tmp -= a[i][j]*x[j];

        //if(tmp%a[i][i] != 0) return -2;
        x[i] = tmp/a[i][i];
    }
    return 0;
}
View Code
double a[maxn][maxn],x[maxn];
int equ,var;
int Gauss()
{
    for(int i=0,j=0;i<equ&&j<var;j++){
        for(int k=i;k<equ;k++){
            if(fabs(a[k][j])>=EPS){
                swap(a[i],a[k]);
                break;
            }
        }
        if(fabs(a[i][j])<EPS) continue;
        for(int k=i+1;k<equ;k++){
            if(fabs(a[k][j])>=EPS){
                double r=a[k][j]/a[i][j];
                for(int p=j;p<=var;p++) a[k][p]-=a[i][p]*r;
            }
        }
        i++;
    }
    REP(i,0,var-1) x[i]=0;
    for(int i=equ-1;i>=0;i--){
        for(int j=0;j<var;j++){
            if(i!=j) x[i]+=a[i][j]*x[j];
        }
        x[i]-=a[i][var];
        x[i]=(-x[i])/a[i][i];
    }
}
View Code

 

posted @ 2016-10-31 16:30  __560  阅读(398)  评论(0编辑  收藏  举报