Cod sursa(job #2147570)

Utilizator Iulia14iulia slanina Iulia14 Data 28 februarie 2018 20:32:10
Problema Tablete Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <fstream>

using namespace std;
ifstream cin ("tablete.in");
ofstream cout ("tablete.out");
int a[1005][1005];
char v[1000005];
int main()
{
    int n,k,i,j,cnt;
    cin>>n>>k;
    if (n%2==0)
    {
        if (k%2==0)
        {
            for (i=1;i<=n;i++)
            {
                for (j=1;j<=n;j++)
                {
                    cout<<j+(i-1)*n<<" ";
                }
                cout<<endl;
            }
        }
        else
        {
            for (i=1;i<=n;i++)
            {
                for (j=1;j<k;j++)
                    cout<<j+(i-1)*n<<" ";
                cout<<n+(i-1)*n<<" ";
                for (j=k+1;j<n;j++)
                    cout<<j+(i-1)*n;
                cout<<k+(i-1)*n<<endl;
            }
        }
    }
    else
    {
        for (i=1;i<=n;i++)
        {
            a[i][k]=2*i;
            v[2*i]=1;
        }
        cnt=0;
        for (i=1;i<=n;i++)
        {
            for (j=1;j<=n;j++){
                if (j!=k)
                {cnt++;
                if (v[cnt]!=0)
                    cnt++;
                a[i][j]=cnt;
                }
            }
        }
        for (i=1;i<=n;i++)
        {
            for (j=1;j<=n;j++)
            {
                cout<<a[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}