Cod sursa(job #3331240)

Utilizator LucaWalucaLuca Munteanu LucaWaluca Data 25 decembrie 2025 23:16:59
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("tablete.in");
ofstream out("tablete.out");
int mat[1005][1005];
int main()
{
    int n,k,nr=0;
    in>>n>>k;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++)
            mat[i][j]=++nr;
    for(int i=1;i<=n;i++)
        for(int j=k+1;j<=n;j++)
                mat[i][j]=++nr;
    if(k%2)
    {
        for(int i=2;i<=n;i+=2)
            swap(mat[i][1],mat[i-1][k]);
        if(n%2)
            swap(mat[n][k],mat[1][k+1]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
            out<<mat[i][j]<<" ";
        out<<"\n";
    }
    return 0;
}