Pai nai ce face:
#include <bits/stdc++.h>
using namespace std;
int n,k,a[25][25];
int main()
{
cin >> n >> k;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(i<=k and j<=k) a[i][j]=1;
else if(i<=k and j>k) a[i][j]=2;
else if(i>k and j<=k) a[i][j]=3;
else a[i][j]=4;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
cout << a[i][j] << " ";
cout << endl;
}
return 0;
}