[模板] 高斯消元

[模板] 高斯消元

由于acwing的模板只针对$n\cdot n$的方程有唯一解有效果，这里贴出来对于任意矩阵的模板，以及枚举自由元的方法。

/*
n行m列，时间复杂度O(n^2*m)
*/
//从后往前枚举每一个自由变量的值，遇到了主元列就计算，然后把行上移一格
int num[maxn+10],start[maxn+100];
void dfs(int r/*第r行*/, int c/*第c列*/, int sum/*和*/){
    if(sum > ans) return;//剪枝
    if(!r){
        ans = min(ans,sum);
        return;
    }
    if(c == start[r]){//在主元列
        num[c] = a[r][n+1];
        for(int i = c+1; i <= n; i++) num[c] -= num[i]*a[r][i];
        dfs(r-1,c-1,sum+num[c]);
    }
    else{
        num[c] = 0; dfs(r,c-1,sum);
        num[c] = 1; dfs(r,c-1,sum+num[c]);
    }
}

void gauss(){
    //普通消元+判断无解
    //debug(1);
    int r = 1;
    memset(start,0,sizeof start);
    for(int c = 1; c <= n; c++){
        int t = r;
        for(int i = r; i <= n; i++) if(a[i][c]){ t = i; break; }
        if(!a[t][c]) continue;
        for(int i = n+1; i >= c; i--) swap(a[r][i],a[t][i]);
        start[r] = c;//标记主元列
        for(int i = r+1; i <= n; i++){
            if(a[i][c])
            for(int j = n+1; j >= c; j--) 
                a[i][j] ^= a[i][c]&a[r][j];
        }
        r++;
    }
    //debug(2);
    if(r <= n){
        for(int i = r; i <= n; i++) if(a[i][n+1]) return;
    }
    r--;
    /*//注意，对于需要枚举自由元的消元中，不能加入这一步，会报错。
    //其他情况最好加上。
	for(int i = r; i >= 1; i--)
		for(int j = i-1; j >= 1; j--)
			a[j][n+1] -= a[j][i]*a[i][n+1]; 
	*/
    dfs(r,n,0);
}