Cod sursa(job #2260326)

Utilizator radubigColtos Radu radubig Data 14 octombrie 2018 20:29:56
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <fstream>

using namespace std;
ifstream in("tablete.in");
ofstream out("tablete.out");
int n,a[1001][1001],k,x,q,i,j;
int main()
{
    in>>n>>k;
    if(n%2==1 && k%2==1) x=n*k+1-2*(n-1);
    else x=n*k-2*(n-1);
    q=x;
    for(i=1;i<=n;i++)
    {
        a[i][k]=q;
        q+=2;
    }
    q=0;
    for(i=1;i<=n;i++) for(j=1;j<k;j++)
    {
        q++;
        if(q>=x && q%2==0) q++;
        a[i][j]=q;
    }
    q++;
    for(i=1;i<=n;i++) for(j=k+1;j<=n;j++)
    {
        q++;
        a[i][j]=q;
        if(n%2==1 && k%2==1 && i==1 && j==k+1) q++;
    }
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++) out<<a[i][j]<<' ';
        out<<'\n';
    }
    return 0;
}