#981379 (C++17) No.2958 Placing Many L-s

• 問題: No.2958 Placing Many L-s
• ユーザー: 👑 binap
• 提出日時: 2024-05-13 09:11:15
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 38 ms / 2,000 ms
• コード長: 1,877 bytes
• コンパイル時間: 4,595 ms
• コンパイル使用メモリ: 267,760 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-08 20:50:28
• 合計ジャッジ時間: 7,813 ms
• ジャッジサーバーID （参考情報）: judge4 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	24 ms; 5,248 KB
testcase_04	AC	25 ms; 5,248 KB
testcase_05	AC	27 ms; 5,248 KB
testcase_06	AC	1 ms; 5,248 KB
testcase_07	AC	4 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	4 ms; 5,248 KB
testcase_10	AC	3 ms; 5,248 KB
testcase_11	AC	7 ms; 5,248 KB
testcase_12	AC	4 ms; 5,248 KB
testcase_13	AC	5 ms; 5,248 KB
testcase_14	AC	4 ms; 5,248 KB
testcase_15	AC	17 ms; 5,248 KB
testcase_16	AC	5 ms; 5,248 KB
testcase_17	AC	7 ms; 5,248 KB
testcase_18	AC	1 ms; 5,248 KB
testcase_19	AC	24 ms; 5,248 KB
testcase_20	AC	23 ms; 5,248 KB
testcase_21	AC	2 ms; 5,248 KB
testcase_22	AC	25 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	AC	38 ms; 5,248 KB
testcase_25	AC	29 ms; 5,248 KB
testcase_26	AC	30 ms; 5,248 KB
testcase_27	AC	31 ms; 5,248 KB
testcase_28	AC	32 ms; 5,248 KB
testcase_29	AC	8 ms; 5,248 KB
testcase_30	AC	2 ms; 5,248 KB

ソースコード

#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
using namespace atcoder;

int solve(){
	int n, m;
	cin >> n >> m;
	bool FLIP = false;
	auto v2 = [&](int x){
		int res = 0;
		while(x % 2 == 0){
			res++;
			x /= 2;
		}
		return res;
	};
	if(v2(n) + v2(m) <= 2){
		cout << "-1\n";
		return 0;
	}
	if(v2(n) > v2(m)){
		swap(n, m);
		FLIP = true;
	}
	
	int cnt = 0;
	vector<vector<int>> a(n, vector<int>(m));
	vector<vector<int>> B = {{1, 1, 1, 2}, {1, 2, 2, 2}};
	auto set_B = [&](int si, int sj){
		rep(di, 2) rep(dj, 4){
			a[si + di][sj + dj] = cnt + B[di][dj];
		}
		cnt += 2;
	};
	vector<vector<int>> C = {{1, 1, 3, 3, 3, 4, 6, 6}, {1, 2, 3, 4, 4, 4, 5, 6}, {1, 2, 2, 2, 5, 5, 5, 6}};
	auto set_C = [&](int si, int sj){
		rep(di, 3) rep(dj, 8){
			a[si + di][sj + dj] = cnt + C[di][dj];
		}
		cnt += 6;
	};
	
	
	// case (II)
	if(v2(n) >= 1 and v2(m) >= 2){
		int k = n * m / 4;
		rep(i, n / 2) rep(j, m / 4){
			set_B(i * 2, j * 4);
		}
		cout << k << "\n";
		if(FLIP){
			vector<vector<int>> a_old(m, vector<int>(n));
			swap(a, a_old);
			rep(i, n) rep(j, m) a[j][i] = a_old[i][j];
			swap(n, m);
		}
		rep(i, n){
			rep(j, m) cout << a[i][j] << (j == m - 1 ? "\n" : " ");
		}
		return 0;
	}
	// case (III)
	if(v2(n) == 0 and v2(m) >= 3){
		if(n == 1){
			cout << "-1\n";
			return 0;
		}
		int k = n * m / 4;
		int n2 = n / 2 - 1;
		int m2 = m / 4;
		rep(i, n / 2 - 1) rep(j, m / 4){
			set_B(i * 2, j * 4);
		}
		rep(j, m / 8){
			set_C(n - 3, j * 8);
		}
		cout << k << "\n";
		if(FLIP){
			vector<vector<int>> a_old(m, vector<int>(n));
			swap(a, a_old);
			rep(i, n) rep(j, m) a[j][i] = a_old[i][j];
			swap(n, m);
		}
		rep(i, n){
			rep(j, m) cout << a[i][j] << (j == m - 1 ? "\n" : " ");
		}
		return 0;
	}
	
	assert(false);
	return 0;
}

int main(){
	int t;
	cin >> t;
	for(int i = 0; i < t; i++) solve();
}