Cod sursa(job #2029613)

Utilizator cont_nouCont Nou cont_nou Data 30 septembrie 2017 12:07:09
Problema Tablete Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.4 kb
#include <iostream>
#include <fstream>
using namespace std;
int a[1001][1001],n,k,nr,uv;
int main()
{
    ifstream f("tablete.in");
    ofstream g("tablete.out");
    nr=1;
    f>>n>>k;
    if(k%2==0)
    {
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
            {
                a[i][j]=nr;
                nr++;
            }

        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                g<<a[i][j]<<" ";
            g<<endl;
        }
    }
    else
    {
        uv=0;
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
            {
                if(j==1 && uv!=0)
                {
                    a[i][j]=uv;
                    uv=0;
                }
                else if(j!=k)
                {
                    a[i][j]=nr;
                    nr++;
                }
                else if(j==k && nr%2==0)
                {
                    a[i][j]=nr;
                    nr++;
                }
                else if(j==k && nr%2==1)
                {
                    uv=nr;
                    nr++;
                    a[i][j]=nr;
                    nr++;
                }

            }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                g<<a[i][j]<<" ";
            g<<endl;
        }
    }
    return 0;
}