Cod sursa(job #1882351)

Utilizator tgm000Tudor Mocioi tgm000 Data 17 februarie 2017 09:46:39
Problema Tablete Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include<cstdio>
int a[1001][1001];
int main(){
    int n,k,i,j,pls;
    freopen("tablete.in","r",stdin);
    freopen("tablete.out","w",stdout);
    scanf("%d%d",&n,&k);
    if(k%2==0)
        i=1;
    else
        i=2;
    for(j=1;j<=k;j++,i++)
        a[1][j]=i;
    if(k%2==1){
        a[2][1]=1;
        j=2;
    }else
        j=1;
    for(;j<=k;j++,i++)
        a[2][j]=i;
    for(j=k+1;j<=n;j++,i++)
        a[1][j]=i;
    for(j=k+1;j<=n;j++,i++)
        a[2][j]=i;
    pls=2*n;
    for(i=3;i<n;i+=2,pls+=2*n){
        for(j=1;j<=n;j++){
            a[i][j]=a[(i-1)%2+1][j]+pls;
            a[i+1][j]=a[i%2+1][j]+pls;
        }
    }
    if(n%2==1){
        j=n*(n-1);
        if(k%2==0)
            for(i=1,j++;j<=n*n;j++,i++)
                a[n][i]=j;
        else{
            a[n-1][n]=n*n;
            for(i=1;i<=n;i++,j++)
                a[n][i]=j;
        }
    }
    for(i=1;i<=n;i++){
        for(j=1;j<=n;j++)
            printf("%d ",a[i][j]);
        printf("\n");
    }
    return 0;
}