Cod sursa(job #2029662)

Utilizator cont_nouCont Nou cont_nou Data 30 septembrie 2017 12:34:50
Problema Tablete Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("tablete.in");
ofstream fout("tablete.out");
int a[1000][1000];

int main()
{
    int n,k,i,j,x=1,y=2,ut,aux;
    fin>>n>>k;
    if(k%2==0 && n%2==0)
    {
        for(j=1;j<=n;j++)
            for(i=1;i<=n;i++)
        {
            if(j%2==1) {a[i][j]=x; x+=2;}
            else {a[i][j]=y; y+=2;}
        }
    }
    else if(k%2==0 && n%2==1)
    {
        for(j=1;j<n;j++)
            for(i=1;i<=n;i++)
        {
            if(j%2==1) {a[i][j]=x; x+=2;}
            else {a[i][j]=y; y+=2;}
        }
        ut=n-1;
        for(i=1;i<=n;i++) {a[i][n]=n-ut+(y-2); ut--;}
    }

    else if(k%2==1 && n%2==0)
    {
        for(i=1;i<=n;i++) for(j=1;j<k;j++){a[i][j]=x; x+=1;}
        for(i=1;i<=n;i++) {a[i][k]=x+2*i;}
        x++;
        for(j=k+1;j<=n;j++) for(i=1;i<=n;i++) {a[i][j]=x; x+=2;}
        aux=a[1][k]; a[1][k]=a[n][k-1]; a[n][k-1]=aux;
    }


    for(i=1; i<=n; i++)
    {
        for(j=1; j<=n; j++) fout<<a[i][j]<<" ";
        fout<<"\n";
    }
    return 0;
}