Cod sursa(job #2148577)

Utilizator AnnaLipianuLipianu Ana AnnaLipianu Data 1 martie 2018 20:02:25
Problema Tablete Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int v[1001][1001];
int main()
{
    freopen("tablete.in","r",stdin);
    freopen("tablete.out","w",stdout);
    int n,k,x,i,j;
    cin>>n>>k;

        if(k%2==0)
        {
            if(n%2==0)
            for(i=1;i<=n;i++)
                for(j=1;j<=n;j++)
                    v[i][j]=n*(i-1)+j;
            else
            {
                x=0;
            for(i=1;i<=n;i++)
                for(j=1;j<=k;j++)
                {
                    x++;
                    v[i][j]=x;
                }
            for(i=1;i<=n;i++)
                for(j=k+1;j<=n;j++)
                {
                    x++;
                    v[i][j]=x;
                }
            }
        }
        else
        {
            x=0;
         for(i=1;i<=n;i++)
            for(j=1;j<=k;j++)
            {
                x++;
                v[i][j]=x;
            }
        for(i=1;i<n;i+=2)
        swap(v[i][k],v[i+1][1]);
           for(i=1;i<=n;i++)
            for(j=k+1;j<=n;j++)
            {
                x++;
                v[i][j]=x;
            }
        if(n%2==1)swap(v[n][k],v[1][k+1]);
         }
        for(i=1;i<=n;i++)
            {
                for(j=1;j<=n;j++)
                cout<<v[i][j]<<" ";
                cout<<'\n';
            }
    return 0;
}