Cod sursa(job #2611717)

Utilizator Fantastic_Mantudor voicu Fantastic_Man Data 7 mai 2020 14:50:12
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.56 kb
#include <fstream>
using namespace std;
ifstream fin("tablete.in");
ofstream fout("tablete.out");
int a[1001][1001];
int main() {
    int n,k,nr=0,i,j;
    fin>>n>>k;
    for(i=1;i<=n;i++)
        for(j=1;j<=k;j++)
            a[i][j]=++nr;
    for(i=1;i<=n;i++)
        for(j=k+1;j<=n;j++)
            a[i][j]=++nr;
    if(k%2==1) {
        for(i=2;i<=n;i+=2)
            swap(a[i][1],a[i-1][k]);
        if(n%2==1)
            swap(a[n][k],a[1][k+1]);
    }
    for(i=1;i<=n;i++,fout<<'\n')
        for(j=1;j<=n;j++)
            fout<<a[i][j]<<' ';
    
    return 0;
}