Cod sursa(job #983332)

Utilizator dariusdariusMarian Darius dariusdarius Data 11 august 2013 15:59:48
Problema Tablete Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 kb
#include<cstdio>
#include<algorithm>
#include<cassert>
using namespace std;
int a[1005][1005];
void solve_kpar(int n,int k)
{
    int nr=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++)
            a[i][j]=nr++;
    for(int i=1;i<=n;i++)
        for(int j=k+1;j<=n;j++)
            a[i][j]=nr++;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            printf("%d%c",a[i][j],j==n?'\n':' ');
}
void solve_2l(int i,int n,int k)
{
    int nr=(i-1)*k;
    for(int j=1;j<=k;j++)
        a[i][j]=++nr;
    for(int j=1;j<=k;j++)
        a[i+1][j]=++nr;
    a[i][k]++;
    a[i+1][1]--;
}
void solve_npar(int n,int k)
{
    for(int i=1;i<=n;i+=2)
        solve_2l(i,n,k);
    int nr=n*k+1;
    for(int i=1;i<=n;i++)
        for(int j=k+1;j<=n;j++)
            a[i][j]=nr++;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            printf("%d%c",a[i][j],j==n?'\n':' ');
}
void solve_nimpar(int n,int k)
{
    for(int i=1;i<n;i+=2)
        solve_2l(i,n,k);
    for(int i=1;i<k;i++)
        a[n][i]=(n-1)*k+i;
    a[n][k]=n*k+1;
    int nr=n*k+1;
    for(int i=1;i<=n;i++)
        for(int j=k+1;j<=n;j++)
            a[i][j]=nr++;
    a[1][k+1]=n*k;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            printf("%d%c",a[i][j],j==n?'\n':' ');
}
int main()
{
    freopen("tablete.in","r",stdin);
    freopen("tablete.out","w",stdout);
    int n,k;
    scanf("%d%d",&n,&k);
    if(k%2==0) solve_kpar(n,k);
    else
        if(n%2==0)
            solve_npar(n,k);
        else
            solve_nimpar(n,k);
    return 0;
}