Cod sursa(job #2229789)

Utilizator FunnyStockyMihnea Andreescu FunnyStocky Data 8 august 2018 09:40:00
Problema Tablete Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <fstream>
#include <algorithm>

using namespace std;

ifstream fin("tablete.in");
ofstream fout("tablete.out");

const int N=1000+5;
int v[N][N],n;
int k;

int main() {
    fin>>n>>k;
    if(k%2==0) {
        int id=0;
        for(int i=1;i<=n;i++) {
            for(int j=1;j<=k;j++) {
                v[i][j]=++id;
            }
        }
        for(int i=1;i<=n;i++) {
            for(int j=k+1;j<=n;j++) {
                v[i][j]=++id;
            }
        }
    }
    else {
        int id=0;
        for(int i=1;i<=n;i++) {
            for(int j=1;j<=k;j++) {
                v[i][j]=++id;
            }
        }
        for(int i=1;i<n;i+=2) {
            swap(v[i][k],v[i+1][1]);
        }
        for(int i=1;i<=n;i++) {
            for(int j=k+1;j<=n;j++) {
                v[i][j]=++id;
            }
        }
        if(n%2) {
            swap(v[n][k],v[1][k+1]);
        }
    }
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            fout<<v[i][j]<<" ";
        }
        fout<<"\n";
    }
    return 0;
}
/**


**/