【高斯消元解异或方程组】[CQOI2014][HYSBZ/BZOJ3503]和谐矩阵

题目
分析:一个元素只可能被它上下左右即自己影响,设mat[i][j]为当且元素的值。那么mat[i][j]^mat[i+1][j]^mat[i][j+1]^mat[i-1][j]^mat[i][j-1]=0
所以我们可以据此列出m*n个方程,根据高斯消元求解即可。
但是,此题答案显然不唯一,如何确定答案呢?
从最后一行开始,我们假设这一行有cnt个元素系数不唯一,那我们将前面cnt-1个视为自由元,赋值为1,然后求出第cnt个元素的值,在向上一行回代,然后继续求解即可。
下面,我用两种算法解决这道题目。
高斯约当消元法:

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 40
int a[MAXN*MAXN+10][MAXN*MAXN+10],n,m,dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}},var,equ,x[MAXN*MAXN+10];
bool vis[MAXN*MAXN+10];
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
void read(){
    Read(n),Read(m);
    int i,j,d;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++){
            for(d=0;d<4;d++)
                if(i+dir[d][0]&&i+dir[d][0]<=n&&j+dir[d][1]&&j+dir[d][1]<=m)
                    a[(i-1)*m+j][(i+dir[d][0]-1)*m+j+dir[d][1]]=1;
                a[(i-1)*m+j][(i-1)*m+j]=1;
        }
    equ=var=n*m;
}
void gauss_jordan(){
    int row,col,i,j,cnt;
    for(row=col=1;row<=equ&&col<=var;row++,col++){
        if(!a[row][col])
            for(i=row+1;i<=equ;i++)
                if(a[i][col]){
                    swap(a[row],a[i]);
                    break;
                }
        if(!a[row][col]){
            row--;
            continue;
        }
        for(i=1;i<=equ;i++)
            if(i!=row&&a[i][col])
                for(j=var+1;j>=col;j--)
                    a[i][j]^=a[row][j];
    }
    row--;
    for(i=row;i;i--){
        cnt=0;
        for(j=i;j<=var;j++)
            if(a[i][j]){
                if(!vis[j])
                    cnt++;
                else
                    a[i][var+1]^=x[j],a[i][j]=0;
            }
        for(j=i;j<=var;j++)
            if(a[i][j]){
                if(--cnt)
                    x[j]=1,a[i][var+1]^=1;
                else
                    x[j]=a[i][var+1];
                vis[j]=1;
            }
    }
}
void print(){
    int i,j;
    for(i=1;i<=n;i++){
        for(j=1;j<=m;j++)
            printf("%d ",x[(i-1)*m+j]);
        puts("");
    }
}
int main()
{
    read();
    gauss_jordan();
    print();
}

上面这个是高斯约当消元法,然而经测试,普通高斯消元更快(4000+ms->400+ms)。这道题因为要回代,没有必要用高斯约当消元法。
两种方法就差一个循环变量啦啦啦,真是差之毫厘,谬以千里
高斯消元法:

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 40
int a[MAXN*MAXN+10][MAXN*MAXN+10],n,m,dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}},var,equ,x[MAXN*MAXN+10];
bool vis[MAXN*MAXN+10];
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
void read(){
    Read(n),Read(m);
    int i,j,d;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++){
            for(d=0;d<4;d++)
                if(i+dir[d][0]&&i+dir[d][0]<=n&&j+dir[d][1]&&j+dir[d][1]<=m)
                    a[(i-1)*m+j][(i+dir[d][0]-1)*m+j+dir[d][1]]=1;
                a[(i-1)*m+j][(i-1)*m+j]=1;
        }
    equ=var=n*m;
}
void gaussian_elimination(){
    int row,col,i,j,cnt;
    for(row=col=1;row<=equ&&col<=var;row++,col++){
        if(!a[row][col])
            for(i=row+1;i<=equ;i++)
                if(a[i][col]){
                    swap(a[row],a[i]);
                    break;
                }
        if(!a[row][col]){
            row--;
            continue;
        }
        for(i=row+1;i<=equ;i++)
            if(a[i][col])
                for(j=var+1;j>=col;j--)
                    a[i][j]^=a[row][j];
    }
    row--;
    for(i=row;i;i--){
        cnt=0;
        for(j=i;j<=var;j++)
            if(a[i][j]){
                if(!vis[j])
                    cnt++;
                else
                    a[i][var+1]^=x[j],a[i][j]=0;
            }
        for(j=i;j<=var;j++)
            if(a[i][j]){
                if(--cnt)
                    x[j]=1,a[i][var+1]^=1;
                else
                    x[j]=a[i][var+1];
                vis[j]=1;
            }
    }
}
void print(){
    int i,j;
    for(i=1;i<=n;i++){
        for(j=1;j<=m;j++)
            printf("%d ",x[(i-1)*m+j]);
        puts("");
    }
}
int main()
{
    read();
    gaussian_elimination();
    print();
}

你可能感兴趣的:(C++,高斯消元,异或方程组,CQOI,bzoj)