Cod sursa(job #231299)

Utilizator georgianaGane Andreea georgiana Data 14 decembrie 2008 12:04:24
Problema Tablete Scor 100
Compilator cpp Status done
Runda Algoritmiada 2009, Runda 1, Clasele 9-10 Marime 0.64 kb
#include <fstream>
using namespace std;

int n, k, a[1001][1001];

int main()
{
	ifstream in("tablete.in");
	ofstream out("tablete.out");
	in >> n >> k;
	k--;
	int act = n*n, ram = 0;
	for (int i = 0; i < n; i++){
		for (int j = n-1; j > k; j--)
			if (ram) a[i][j] = ram, ram = 0;
			else a[i][j] = act--;
		if (act % 2 == 0) a[i][k] = act--;
		else ram = act--, a[i][k] = act--;
	}
	for (int i = 0; i < n; i++)
		for (int j = k-1; j >= 0; j--)
			if (ram) a[i][j] = ram, ram = 0;
			else a[i][j] = act--;
	for (int i = 0; i < n; i++){
		out << a[i][0];
		for (int j = 1; j < n; j++)
			out << " " << a[i][j];
		out << "\n";
	}
	out.close();
	return 0;
}