Cod sursa(job #2399226)

Utilizator darabana.raresDarabana Rares Cristian darabana.rares Data 7 aprilie 2019 10:15:59
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("tablete.in");
ofstream out("tablete.out");
int m,n,i,j,k,numar,v[1005][1005],val,c;
int main()
{
    in>>n>>k;
     if(k%2==0)
     {
         val=1;
         for(i=1;i<=n;i++)
            for(j=1;j<=k;j++)
         {
             v[i][j]=val;
             val++;
         }
         for(i=1;i<=n;i++)
            for(j=k+1;j<=n;j++)
         {
             v[i][j]=val;
             val++;
         }
     }
     else
     {
         int val=1;
         for(i=1;i<=n;i++)
            for(j=1;j<=k;j++)
         {
             if(v[i][j]!=0)continue;
             if(j==k)
             {
                 if(val%2==1)
                 {
                     if(i==n)v[1][k+1]=val;
                     else v[i+1][1]=val;
                     val++;
                 }
                 v[i][j]=val;
             }
             else v[i][j]=val;
             val++;
         }
         for(i=1;i<=n;i++)
            for(j=k+1;j<=n;j++)
         {
             if(v[i][j]!=0)continue;
             v[i][j]=val;
             val++;
         }
     }
    for(i=1;i<=n;i++)
        {for(j=1;j<=n;j++)out<<v[i][j]<<" ";
        out<<'\n';}
    return 0;
}