Cod sursa(job #2420026)

Utilizator mjmilan11Mujdar Milan mjmilan11 Data 10 mai 2019 10:39:24
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("tablete.in");
ofstream fout("tablete.out");

const int NMAX = 1005;
int rasp[NMAX][NMAX];

int main()
{
    int n,k;
    fin >> n >> k;
    int ind=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=k;j++)
        {
            ind++;
            rasp[i][j]+=ind;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=k+1;j<=n;j++)
        {
            ind++;
            rasp[i][j]+=ind;
        }
    }
    if(k%2==1)
    {
        for(int i=1;i<=n;i++)
        {
            if(i%2==0) rasp[i][1]--;
            else rasp[i][k]++;
        }
        if(n%2==1) rasp[1][k+1]--;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            fout << rasp[i][j] << ' ';
        }
        fout << '\n';
    }
    return 0;
}