【CodeForces 625C】K-special Tables

#include<cstdio>
int n,k;
int a[505][505];
int main()
{
    scanf("%d%d",&n,&k);
    int m=1;

    for(int i=1; i<=n; i++)
        for(int j=1; j<k; j++)
        {
            a[i][j]=m++;
        }
    for(int i=1; i<=n; i++)
        for(int j=k; j<=n; j++)
        {
            a[i][j]=m++;
        }
    printf("%d\n",(n*(n-2+k)+k+1)*n/2);
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            printf("%d ",a[i][j]);
        }
        printf("\n");
    }
    return 0;
}