codeforces 432E Square Tiling

codeforces 432E Square Tiling

题意

题解

代码

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define rep(i, a, b) for(int i=(a); i<(b); i++)
#define sz(x) (int)x.size()
#define de(x) cout<< #x<<" = "<<x<<endl
#define dd(x) cout<< #x<<" = "<<x<<" "
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;

const int N = 111;

int n, m;
bool in[N][N];
int ans[N][N], ban[N][N];

int calc(int x) {
	for(int i = 0; ; ++i) if(!(x>>i&1)) return i;
}

int main() {
	std::ios::sync_with_stdio(false);
	std::cin.tie(0);
	cin >> n >> m;
	rep(i, 1, n+1) rep(j, 1, m+1) in[i][j] = 1;
	memset(ans, -1, sizeof(ans));
	rep(i, 1, n+1) rep(j, 1, m+1) if(ans[i][j]==-1) {
		int c = calc(ban[i][j]), k;
		rep(t, 1, n+1) {
			bool ok = 1;
			if(i+t-1>n||j+t-1>m) ok = 0;
			rep(x, i, i+t) if(ans[x][j+t-1]!=-1 || calc(ban[x][j+t-1])>c) ok = 0;
			rep(y, j, j+t) if(ans[y][i+t-1]!=-1 || calc(ban[y][i+t-1])>c) ok = 0;
			if(!ok) break;
			k = t;
			if(j+t<=m&&c>calc(ban[i][j+t])) break;
		}
		rep(x, i, i+k) rep(y, j, j+k) ans[x][y] = c;
		rep(x, i, i+k) ban[x][j-1] |= (1<<c), ban[x][j+k] |= (1<<c);
		rep(y, j, j+k) ban[i-1][y] |= (1<<c), ban[i+k][y] |= (1<<c);
	}
	rep(i, 1, n+1) {
		rep(j, 1, m+1) cout << (char)(ans[i][j]+'A');
		cout << endl;
	}
	return 0;
}