Cod sursa(job #3332111)

Utilizator tileadavidtileadavid tileadavid Data 4 ianuarie 2026 14:18:34
Problema Tablete Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>

using namespace std;

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

int a[1005][1005];

int main()
{
    int n, k;
    cin >> n >> k;
    int x = 0;

    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= k; ++j){
            a[i][j] = ++x;
        }
    }
    for (int i = 1; i < n; ++i){
        if (a[i][k] % 2 == 1){
            swap(a[i][k], a[i + 1][1]);
        }
    }
    if (n % 2 == 0){
        ++x;
    }
    for (int i = 1; i <= n; ++i){
        for (int j = k + 1; j <= n; ++j){
            a[i][j] = x;
            ++x;
        }
    }

    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= n; ++j){
            cout << a[i][j] << ' ';
        }
        cout << '\n';
    }

    return 0;
}