Cod sursa(job #2985213)

Utilizator Zed1YasuoAlex Birsan Zed1Yasuo Data 25 februarie 2023 22:18:34
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <fstream>

using namespace std;
ifstream f("tablete.in");
ofstream g("tablete.out");
int n,k,a[1005][1005];
bool fr[1000005];
int main()
{
    f>>n>>k;
    int r=n*n;
    for(int j=n;j>k;j--)
    {
        for(int i=n;i>0;i--)
            a[i][j]=r,r--;
    }
    if(r%2==1)
    {
        a[1][k+1]=r;
        r++;
    }
    for(int i=n;i>0;i--)
        a[i][k]=r,fr[r]=true,r-=2;
    r=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<k;j++)
        {
            while(fr[r])
                r++;
            fr[r]=true;
            a[i][j]=r;
        }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
            g<<a[i][j]<<" ";
        g<<'\n';
    }
    return 0;
}