Cod sursa(job #3165619)

Utilizator MariosulmarioMario Badea Mariosulmario Data 6 noiembrie 2023 16:51:07
Problema Tablete Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <iostream>
#include <fstream>
using namespace std;

int cioco[1002][1002];
ifstream in("tablete.in");
ofstream out("tablete.out");

int main()
{
    int n,k;
    in>>n>>k;
    long long nr=1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=k;j++){
            cioco[i][j] = nr;
            nr++;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=k+1;j<=n;j++){
            cioco[i][j] = nr;
            nr++;
        }
    }
    if(k%2==1){
        for(long long i=1;i<n;i=i+2){
            long long aux = cioco[i][k];
            cioco[i][k] = cioco[i+1][1];
            cioco[i+1][1] = aux;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            out<<cioco[i][j];
            out<<" ";
        }
        out<<"\n";
    }
}