Cod sursa(job #2061092)

Utilizator Bogdan_BuzatuBuzatu Bogdan Mihai Bogdan_Buzatu Data 8 noiembrie 2017 22:04:50
Problema Tablete Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.14 kb
#include <fstream>
using namespace std;
ifstream fin("tablete.in");
ofstream fout("tablete.out");

int n, k, i, j, s, v[1002][1002],a;

int main () {
    fin>>n>>k;
    if(n%2==0 && k%2==0){
        for(i=1; i<=n; i++){
            for(j=1; j<=n; j++){
                v[i][j]=(i-1)*n+j;
            }

        }
    }
    if(n%2==1 && k%2==0){
        s=n;
        for(i=1; i<=n; i++){
            for(j=1; j<=n; j++){
                if(j==1){

                    v[i][j]=i;
                }
                else{
                    s++;
                    v[i][j]=s;
                }
            }

        }
    }

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

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

 return 0;
}