Cod sursa(job #2259844)

Utilizator FrostfireMagirescu Tudor Frostfire Data 13 octombrie 2018 20:32:16
Problema Tablete Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <fstream>
using namespace std;
ifstream f("tablete.in");
ofstream g("tablete.out");
int n,k,i,j,a[1010][1010],nr;
bool b[1010];
int main()
{
    f>>n>>k;
    if(k%2==0) for(i=1;i<=n;i++) a[i][k] = k*i, b[i*k]=1;
      else { for(i=1;i<=n;i++) { if(i%2==0) a[i][k] = k*i, b[i*k] = 1;
                                    else a[i][k] = k*i+1, b[i*k+1]=1;
                               }
           }
    nr = 1;
    for(i=1;i<=n;i++) for(j=1;j<k;j++) { while(b[nr]) nr++;
                                         a[i][j] = nr;
                                         nr++;
                                       }
    for(i=1;i<=n;i++) for(j=k+1;j<=n;j++) { while(b[nr]) nr++;
                                            a[i][j] = nr;
                                            nr++;
                                          }
    for(i=1;i<=n;i++) { for(j=1;j<=n;j++) g<<a[i][j]<<' ';
                        g<<'\n';
                      }
    return 0;
}