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

• 問題: No.2958 Placing Many L-s
• ユーザー: 👑 binap
• 提出日時: 2024-05-13 06:28:03
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,801 bytes
• コンパイル時間: 4,464 ms
• コンパイル使用メモリ: 265,328 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-08 20:50:23
• 合計ジャッジ時間: 9,914 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	2 ms; 5,248 KB
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	AC	2 ms; 5,248 KB
testcase_07	WA	-
testcase_08	AC	2 ms; 5,248 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	AC	2 ms; 5,248 KB
testcase_19	WA	-
testcase_20	WA	-
testcase_21	AC	2 ms; 5,248 KB
testcase_22	WA	-
testcase_23	AC	3 ms; 5,248 KB
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-

ソースコード

#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 main(){
	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);
		}
		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;
}