Cod sursa(job #2056094)

Utilizator FunnyStockyMihnea Andreescu FunnyStocky Data 4 noiembrie 2017 08:46:11
Problema Tablete Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.18 kb
#include <fstream>
using namespace std;
ifstream cin("tablete.in");
ofstream cout("tablete.out");
int n,k,y=0;
int main()
{
    cin>>n>>k;
    if(n%2==0)
    {
        if(k%2==0)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                    cout<<(i-1)*n+j<<" ";
                cout<<"\n";
            }
        }
        for(int r=1;r<=n;r+=2)
        {
            y++;
            for(int i=1;i<=k;i++)
                cout<<++y<<" ";
            int yn=y+n-1;
            for(int i=k+1;i<=n;i++)
                cout<<++yn<<" ";
            cout<<"\n"<<(r-1)*n+1<<" ";
            for(int i=2;i<=n;i++)
                cout<<++y<<" ";
            cout<<"\n";
            y++;
        }
        return 0;
    }
    if(k%2==0)
    {
        y=n;
        for(int r=1;r<=n;r++)
        {
            cout<<r<<" ";
            for(int c=2;c<=n;c++)
                cout<<++y<<" ";
            cout<<"\n";
        }
        return 0;
    }
    y=n;
    for(int r=1;r<=n;r++)
    {
        for(int c=1;c<n;c++)
            cout<<++y<<" ";
        cout<<r<<" ";
        cout<<"\n";
    }
    return 0;
}