Cod sursa(job #2659692)

Utilizator NeacsuMihaiNeacsu Mihai NeacsuMihai Data 17 octombrie 2020 12:14:11
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream fin ("tablete.in");
ofstream fout ("tablete.out");
int main()
{
    int n, k, i, j;
    fin>>n>>k;

    if(k%2==0)
    {
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=k; j++)
            {
                fout<<(i-1)*k+j<<' ';
            }
            for(j=k+1; j<=n; j++)
            {
                fout<<n*k+(i-1)*(n-k) + j-k<<' ';
            }

            fout<<"\n";
        }
    }
    else
    {
        for(i=1; i<=n; i++)
        {
            if(i%2==0) fout<<(i-1)*k<<' ';
            else fout<<(i-1)*k+1<<' ';
            for(j=2; j<k; j++) fout<<(i-1)*k+j<<' ';
            if(i%2==1) fout<<i*k+1<<' ';
            else fout<<i*k<<' ';

            if(n%2==1 && i==1 && k+1<=n)
            {
                fout<<(n*k)<<' ';
                for(j=k+2; j<=n; j++) fout<<k*n+(i-1)*(n-k)+(j-k)<<' ';
            }
            else for(j=k+1; j<=n; j++)
            {
                fout<<k*n+(i-1)*(n-k)+(j-k)<<' ';
            }
            fout<<"\n";
        }
    }
}