Cod sursa(job #1558874)

Utilizator iulianrotaruRotaru Gheorghe-Iulian iulianrotaru Data 29 decembrie 2015 18:31:01
Problema Tablete Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.55 kb
#include <fstream>
#include <sstream>
using namespace std;
ifstream f("tablete.in");
ofstream g("tablete.out");
int i,j,nr,k,n,v[1012][1012];
stringstream rez;
int main()
{
    f>>n>>k;
    for(i=1;i<=n;++i)
        for(j=1;j<=k;++j) v[i][j]=++nr;
    for(i=1;i<=n;++i)
        for(j=k+1;j<=n;++j) v[i][j]=++nr;
    if(k&1)
    {
        for(i=2;i<=n;i+=2) swap(v[i][1],v[i-1][k]);
        if(n&1) swap(v[n][k],v[1][k+1]);
    }
    for(i=1;i<=n;++i,rez<<'\n')
        for(j=1;j<=n;++j) rez<<v[i][j]<<" ";
    g<<rez.str();
    return 0;
}