Cod sursa(job #2316602)

Utilizator dragos99Homner Dragos dragos99 Data 11 ianuarie 2019 23:31:07
Problema Tablete Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.67 kb
#include<fstream>
using namespace std;
    ifstream f("tablete.in");
    ofstream g("tablete.out");
long v[1001][1001], n, k;
void printMatrix()
{
    long i, j;
    for(i = 1 ; i <= n ; i++){
        for(j = 1 ; j <= n ; j++)
            g<<v[i][j]<<" ";
        g<<'\n';
    }
}
int main()
{
    long nr = 1, i, j;
    f>>n>>k;
    if(k % 2 == 0){
        for(i = 1 ; i <= n ; i++)
            for(j = 1 ; j <= k ; j++){
                v[i][j] = nr;
                nr++;
        }
        for(i = 1 ; i <= n ; i++)
            for(j = k + 1 ; j <= n ; j++){
                v[i][j] = nr;
                nr++;
        }
    }

    if(k % 2 == 1){
            for(i = 1  ; i <= n ; i++){
                for(j = 1 ; j <= k - 1 ; j++){
                    v[i][j] = nr;
                    nr++;
                }
                v[i][j] = nr + 1;
                i++;
                v[i][1] = nr;
                nr += 2;
                for(j = 2 ; j <= k ; j++){
                    v[i][j] = nr;
                    nr ++;
                }
            }
            for(i = 1 ; i <= n ; i++){
                for(j = k + 1 ; j <= n ; j++){
                    if(n % 2 == 1){
                        if(i == 1 && j == k + 1){
                            nr -=4;
                            v[i][j] = nr;
                            nr += 2;
                        }
                        else{
                            v[i][j] = nr;
                            nr++;
                        }
                    }
                }
            }


        if(n % 2 == 1){

        }
    }

    printMatrix();

    return 0;
}