[gotoac]高斯消元

int gcd(int a,int b){
    if(a>b) swap(a,b);
    while(b){int t=b;b=a%b,a=t;}
    return a;
}
  
int lcm(int a,int b){
    return a/gcd(a,b)*b;
}
  
int gauss(int equ,int var,int a[][M]){
    int i,j,k,col;
    clr(free_x,0);
    for(k=col=0;k<equ && col<var;k++,col++){
        int max_r=k;
        for(i=k+1;i<equ;i++)
            if(abs(a[i][col])>abs(a[max_r][col])) max_r=i;
        if(max_r!=k)
            for(j=col;j<var+1;j++) swap(a[k][j],a[max_r][j]);
        if(!a[k][col]){k--;free_x[col]=1;continue;}
        for(i=k+1;i<equ;i++) if(a[i][col]!=0){//消元,具体代码根据题意
            int LCM=lcm(abs(a[i][col]),abs(a[k][col]));
            int ti=LCM/abs(a[i][col]),tk=LCM/abs(a[k][col]);
            if(a[i][col]*a[k][col]<0) tk=-tk;
            for(j=col;j<var+1;j++)
                a[i][j]=a[i][j]*ti-a[k][j]*tk;
        }
    }
    for(i=k;i<equ;i++)
        if(a[i][var]) return -1;//无解
    if(k<var){//无穷解
        for(i=k-1;i>=0;i--){
            int free_id;
            for(j=0;j<var;j++) if(a[i][j]){
                if(free_x[j]) break;//存在自由变元
                else free_id=j;
            }
            if(j<var) continue;
            int tmp=a[i][var];
            for(j=0;j<col;j++)
                if(a[i][j] && j!=free_id)
                    tmp-=a[i][j]*x[j];
            x[free_id]=tmp/a[i][free_id];//求出变元,具体代码根据题意
        }
        return var-k;//自由变元个数
    }
    for(i=k-1;i>=0;i--){//唯一解
        int tmp=a[i][var];
        for(j=i+1;j<var;j++)
            if(a[i][j]) tmp-=a[i][j]*x[j];
        if(tmp%a[i][i]) return -2;//无整数解
        x[i]=tmp/a[i][i];
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值