Cod sursa(job #2316831)

Utilizator cameliapatileaPatilea Catalina Camelia cameliapatilea Data 12 ianuarie 2019 14:45:06
Problema Tablete Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.76 kb
#include <iostream>
#include <fstream>
using namespace std;

ifstream f("tablete.in");
ofstream g("tablete.out");

int  matrice[1001][1001];
bool marcaj[1000001];
int n, k;
void caz1()
{
    int i, j;
    int nr1 = 0;
    for(i = 1; i <= n; i++)
        for(j = 1; j <= n; j++)
            matrice[i][j] = ++nr1;
}
void caz2()
{
    int i, j;
    int  p = 0;
    int nr2 = n * n;

    matrice[1][k] = k + 1;
    marcaj[matrice[1][k]] = 1;
    for(i = 2; i <= n; i++)
        if(i % 2 != 0)
        {
            matrice[i][k] = matrice[i-1][k] + k - 1;
            marcaj[matrice[i][k]] = 1;
        }
        else {
            matrice[i][k] = matrice[i - 1][k] + k + 1;
            marcaj[matrice[i][k]] = 1;
        }
    for(i = 1;i <= n; i++)
        for(j = 1;j <= k; j++)
            if(!matrice[i][j] )
            {
                ++p;
                while(marcaj[p] != 0 && p <= nr2) ++p;
                matrice[i][j] = p;
            }
    for(i = 1; i <= n; i++)
        for(j = k + 1; j <= n; j++)
            if(!matrice[i][j] )
            {
                ++p;
                while (marcaj[p] != 0 && p <= nr2) ++p;
                matrice[i][j] = p;
            }

}
void caz3()
{
    int i, j;
    int nr3 = 0;
    for( i = 1; i <= n; i++)
        for( j = 1; j <= k; j++)
            matrice[i][j] = ++nr3;

    for(i = 1; i <= n; i++)
        for(j = nr3 + 1; j <= n; j++)
            matrice[i][j] = ++nr3;
}
int main() {


    int i, j;
    f >> n >> k;

    if(n % 2 == 0 && k %2 == 0) caz1();

    if((n % 2 == 0) && k % 2 != 0)
    caz2();
    if((n % 2 != 0) && k % 2 != 0)
   caz2();
    if( n % 2 != 0 && k % 2 == 0)
    caz3();
    for(i = 1; i <= n; i++)
    {
        for( j = 1; j <= n; j++)
            g <<matrice[i][j]<<" ";
        g<<"\n";
    }
    return 0;
}