Cod sursa(job #2792766)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 2 noiembrie 2021 12:00:41
Problema Tablete Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.54 kb
#include<fstream>
using namespace std;
ifstream f("tablete.in");
ofstream g("tablete.out");
int i,j,r,k,n,v[1012][1012];
int main()
{
    for(f>>n>>k,i=1;i<=n;++i)
        for(j=1;j<=k;++j)
            v[i][j]=++r;
    for(i=1;i<=n;++i)
        for(j=k+1;j<=n;++j)
            v[i][j]=++r;
    if(k&1) {
        for(i=2;i<=n;i+=2)
            swap(v[i][1],v[i-1][k]);
        if(n&1)
            swap(v[n][k],v[1][k+1]);
    }
    for(i=1;i<=n;++i,g<<'\n')
        for(j=1;j<=n;++j)
            g<<v[i][j]<<' ';
    return 0;
}