Submission #348426

#	Time	Username	Problem	Language	Result	Execution time	Memory
348426		Kevin_Zhang_TW	Red-blue table (IZhO19_stones)	C++17	100 / 100	41 ms	2412 KiB

stones

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T L, T R) { while (L != R) cerr << *L << " \n"[next(L) == R], ++L; }
#else
#define DE(...) 0
#define debug(...) 0
#endif

const int MAX_N = 1010;

int n, m;
char w[MAX_N][MAX_N];

#define rev(w) (char)('+' ^ '-' ^ (w))

void solve(int n, int m) {
	bool rot = false;

	DE(n, m);

	if (n > m) rot = true, swap(n, m);

	auto gen = [&]() {
		for (int i = 0;i < n;++i) for (int j = 0;j < m;++j)
			w[i][j] = '-';

		int res = m;

		vector<int> cnt(m);

		// height is atmost mxh

		int filwid = m / 2 + 1, needh = n / 2 + 1, mxh = n - needh;

		for (int i = 0, H = 0;i < n;++i) {
			bool todo = true;
			for (int j = 0;j < filwid;++j) 
				if ((++cnt[ (H + j) % m ]) > mxh)
					todo = false;

			if (!todo) break;

			for (int j = 0;j < filwid;++j) 
				w[i][ (H + j) % m ] = '+';

			H = (H + filwid) % m;
			++res;
		}
		return res;
	};

	int cnt = gen();

	cout << cnt << '\n';

	if (rot) {
		swap(n, m);
		for (int i = 0;i < n;++i) {
			for (int j = 0;j < m;++j) 
				cout << rev(w[j][i]);
			cout << '\n';
		}
	}
	else {
		for (int i = 0;i < n;++i) {
			for (int j = 0;j < m;++j)
				cout << w[i][j] ;
			cout << '\n';
		}
	}
}
int32_t main() {
	ios_base::sync_with_stdio(0), cin.tie(0);
//  	for (int i = 1;i <= 4;++i) for (int j = 1;j <= 4;++j)
//  		solve(i, j);
  	int T;
  	cin >> T;
  	while (T--) {
  		int n, m;
  		cin >> n >> m;
  		solve(n, m);
 	}
}

Compilation message (stderr)

stones.cpp: In function 'void solve(int, int)':
stones.cpp:14:17: warning: statement has no effect [-Wunused-value]
   14 | #define DE(...) 0
      |                 ^
stones.cpp:28:2: note: in expansion of macro 'DE'
   28 |  DE(n, m);
      |  ^~

• Subtask 1: 17 / 17
• Subtask 2: 10 / 10
• Subtask 3: 16 / 16
• Subtask 4: 11 / 11
• Subtask 5: 15 / 15
• Subtask 6: 31 / 31