BZOJ3503 [Cqoi2014]和谐矩阵

我们发现确定了一行之后，为使这一行没有不合法情况，下一行就确定了，所以我们可以用第一行的数表示出整个矩阵，而合法当且仅当第n+1行全为0，用n+1行的表示高斯消元出第一行然后递推整个矩阵即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<map>
#include<set>
#include<bitset>
#include<queue>
#include<stack>
using namespace std;
#define MAXN 50
#define MAXM 1010
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
#define ll long long
int n,m;
struct data{
	int x[MAXN];
	friend data operator ^(data x,data y){
		int i;
		data z;
		for(i=1;i<=m;i++){
			z.x[i]=x.x[i]^y.x[i];
		}
		return z;
	}
};
data v[MAXN][MAXN];
int a[MAXN][MAXN];
int b[MAXN][MAXN];
void gs(){
	int i,j,k;
	for(i=1;i<=m;i++){
		if(!a[i][i]){
			for(j=i+1;j<=m;j++){
				if(a[j][i]){
					for(k=1;k<=m+1;k++){
						swap(a[i][k],a[j][k]);
					}
					break;
				}
			}
		}
		if(a[i][i]){
			for(j=1;j<=m;j++){
				if(i!=j&&a[j][i]){
					for(k=1;k<=m+1;k++){
						a[j][k]^=a[i][k];
					}
				}
			}
		}
	}
}
int main(){
	int i,j;
	scanf("%d%d",&n,&m);
	for(i=1;i<=m;i++){
		v[1][i].x[i]=1;
	}
	for(i=2;i<=n+1;i++){
		for(j=1;j<=m;j++){
			v[i][j]=v[i-1][j]^v[i-1][j-1]^v[i-1][j+1]^v[i-2][j];
		}
	}
	for(i=1;i<=m;i++){
		for(j=1;j<=m;j++){
			a[i][j]=v[n+1][i].x[j];
		}
	}
	gs();
	for(i=m;i;i--){
		if(!a[i][i]){
			b[1][i]=1;
		}else{
			b[1][i]=a[i][m+1];
			for(j=i+1;j<=m;j++){
				if(a[i][j]){
					b[1][i]^=b[1][j];
				}
			}
		}
	}
	for(i=2;i<=n;i++){
		for(j=1;j<=m;j++){
			b[i][j]=b[i-1][j]^b[i-1][j-1]^b[i-1][j+1]^b[i-2][j];
		}
	}
	for(i=1;i<=n;i++){
		for(j=1;j<=m;j++){
			printf("%d ",b[i][j]);
		}
		printf("\n");
	}
	return 0;
}

/*

*/