Cod sursa(job #1610010)

Utilizator AndreiTACAndrei Cristian AndreiTAC Data 23 februarie 2016 10:54:25
Problema Tablete Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <fstream>

using namespace std;

int main()
{
    ifstream in("tablete.in");
    ofstream out("tablete.out");
    int i,j,n,k;
    in>>n>>k;
    int a[n][n],nr=n*n;
    for(i=n-1;i>=k;i--)
        {
            for(j=n-1;j>=0;j--)
                {
                    a[j][i]=nr;
                    nr--;
                }
        }
    if(nr%2==1)
        {
            a[j+1][i+1]--;
            nr++;
        }
    for(j=n-1;j>=0;j--)
        {
            a[j][k-1]=nr;
            nr-=2;
        }
    nr=a[n-1][k-1]-1;
    for(j=n-1;j>=0;j--)
        {
            a[j][k-2]=nr;
            nr-=2;
        }
    nr++;
    for(i=k-3;i>=0;i--)
        {
            for(j=n-1;j>=0;j--)
                {
                    a[j][i]=nr;
                    nr--;
                }
        }
    for(i=0;i<n;i++)
        {
            for(j=0;j<n;j++)
                {
                    out<<a[i][j]<<" ";
                }
            out<<"\n";
        }
    return 0;
}