Pagini recente » Cod sursa (job #1380778) | Cod sursa (job #360741) | Cod sursa (job #2499109) | Cod sursa (job #2750311) | Cod sursa (job #870527)
Cod sursa(job #870527)
#include <cstdio>
using namespace std;
int mat[1001][1001];
int n,k;
void show()
{
int i,j;
for(i=1;i<=n;++i,printf("\n"))
for(j=1;j<=n;++j)
printf("%d ",mat[i][j]);
}
int main()
{
freopen("tablete.in","r",stdin);
freopen("tablete.out","w",stdout);
int i,j;
scanf("%d %d\n",&n,&k);
if(~n&1)
{
if(~k&1)
{
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
mat[i][j] = (i-1)*n + j;
}
else
{
mat[1][1] = 1;
mat[2][1] = k;
for(j=2;j<=n;++j)
{
mat[1][j] = mat[1][j-1] + 1;
if(j == k)
mat[1][j]++;
}
for(j=2;j<=n;++j)
mat[2][j] = mat[1][j]+n;
for(i=3;i<=n;++i)
for(j=1;j<=n;++j)
mat[i][j] = mat[i-1][j]+n;
for(j=n;j>=k;--j)
mat[n][j] = mat[n][j-1];
mat[n][k-1]--;
mat[n-1][n]=mat[n][k]+1;
}
}
else
{
if(~k&1)
{
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
mat[i][j] = (i-1)*n + j;
for(i=2;i<=n;i+=2)
{
mat[i-1][n] = mat[i][k];
mat[i][k] = mat[i][k-1];
for(j=k-1;j>1;--j)
mat[i][j] = mat[i][j-1];
mat[i][j]--;
}
}
else
{
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
mat[i][j] = (i-1)*n + j;
for(i=1;i<=n;i+=2)
{
for(j=k;j<n;++j)
mat[i][j]=mat[i][j+1];
mat[i][j]++;
mat[i+1][1] = mat[i][k];
}
for(i=1;i<=n;++i)
mat[n][i]--;
mat[n][k] --;
mat[n-1][n]=mat[n][k]+1;
}
}
show();
return 0;
}