Cod sursa(job #2559439)

Utilizator geo_uwuManolachi George geo_uwu Data 27 februarie 2020 12:27:55
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("tablete.in");
ofstream g("tablete.out");
bool ceva[1000005];
int v[1001][1001],n,k,val;
int main()
{
    ios_base::sync_with_stdio(false);
    f.tie(0);
    g.tie(0);
    f>>n>>k;
    int nr=n*n;
    for (int j=n;j>k;j--)
    {
        for (int i=n;i>=1;i--)
        {
            v[i][j]=nr;
            nr--;
        }
    }
    if (nr%2==0)
    {
        val=nr;
    }
    else
    {
        val=nr+1;
        v[1][k+1]=nr;
        ceva[nr]=1;
    }
    for (int i=n;i>=1;i--)
    {
        v[i][k]=val;
        ceva[val]=1;
        val-=2;
    }
    for (int j=k-1;j>=1;j--)
    {
        for (int i=n;i>=1;i--)
        {
            while (ceva[nr]==1)
            {
                nr--;
            }
            v[i][j]=nr;
            nr--;
        }
    }
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=n;j++)
        {
            g<<v[i][j]<<" ";
        }
        g<<'\n';
    }
    return 0;
}