Ruudukko

#include <iostream>
#include <vector>
#include <algorithm>

const int N = 10;
const int MAX = 100000;

bool has [N*N+1];

int main() {
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(0);

	int n;
	std::cin >> n;

	std::vector<int> brute;

	for (int i = 1; i <= n; ++i) {
		for (int j = 0; j < n; ++j) {
			brute.push_back(i);
		}
	}
	bool found = false;
	for (int j = 0; j < MAX; ++j) {
		std::random_shuffle(brute.begin(), brute.end());
		bool fail = false;
		for (int x = 0; x < n; ++x) {
			int s = 0;
			int e = 0;
			for (int y = 0; y < n; ++y) {
				s += brute[x + y * n];
				e += brute[y + x * n];
			}
			if (has[s]) {
				fail = true;
			}
			has[s] = true;
			if (has[e]) {
				fail = true;
			}
			has[e] = true;
		}
		for (int i = 1; i < n * n; ++i) {
			has[i] = false;
		}
		if (fail) continue;
		found = true;
		break;
	}
	if (found) {
		for (int y = 0; y < n; ++y) {
			for (int x = 0; x < n; ++x) {
				std::cout << brute[x + y * n] << ' ';
			}
			std::cout << '\n';
		}
	} else {
		std::cout << "QAQ\n";
	}
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2

Verdict: RUNTIME ERROR

Test 15

Group: 2

Verdict: RUNTIME ERROR

Test 16

Group: 3

Verdict: ACCEPTED

Test 17

Group: 3

Verdict: ACCEPTED

Test 18

Group: 3

Verdict: RUNTIME ERROR

Test 19

Group: 3

Verdict: RUNTIME ERROR

Test 20

Group: 3

Verdict: RUNTIME ERROR

Test 21

Group: 3

Verdict: RUNTIME ERROR