Cod sursa(job #2482133)

Utilizator davidcotigacotiga david davidcotiga Data 27 octombrie 2019 20:18:00
Problema Tablete Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <fstream>
#include <string.h>

using namespace std;

ifstream cin("tablete.in");
ofstream cout("tablete.out");

int main() {
	int N, K;
	cin >> N >> K;
	if (N % 2 == 0) {
		if (K % 2 == 0) {
			for (int i = 1; i <= N * N; ++i) {
				cout << i << " ";
				if (i % N == 0)
					cout << "\n";
			}
		}
		else {
			for (int i = 2; i <= N * N; ++i) {
				cout << i << " ";
				if ((i - 1) % N == 0)
					cout << "\n";
			}
			cout << 1;
		}
	}
	else {
		for (int i = 0; i < N; ++i) {
			if (i % 2 == 0) {
				cout << N * i + 1 << " " << N * i + K << " ";
				for (int j = 3; j < K; ++j)
					cout << N * i + j << " ";
				cout << N * i + 2 << " ";
				for (int j = K + 1; j <= N; ++j)
					cout << N * i + j << " ";
			}
			if (i % 2 == 1) {
				cout << N * i + K << " " << N * i + 2 << " ";
				for (int j = 3; j < K; ++j)
					cout << N * i + j << " ";
				cout << N * i + 1 << " ";
				for (int j = K + 1; j <= N; ++j)
					cout << N * i + j << " ";
			}
			cout << "\n";
		}

	}
	return 0;
}