Cod sursa(job #2262082)

Utilizator pasoi_stefanPasoi Stefan pasoi_stefan Data 16 octombrie 2018 22:38:51
Problema Tablete Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.08 kb
#include<fstream>
#include<algorithm>
using namespace std;
ifstream cin("tablete.in");
ofstream cout("tablete.out");
int n,k,a[1005][1005];
int main(){
    cin>>n>>k;
    if(n%2==0){
        for(int i=1,t=1;i<=n;i++)
            for(int j=1;j<=n;j++,t++)
                a[j][i]=t;
        for(int i=1;i<=n-1;i+=2)
            a[i][k]^=a[i+1][k-1]^=a[i][k]^=a[i+1][k-1];
    }
    else{
        int t=1;
        for(int i=1;i<n;i++)
            for(int j=1;j<n;j++,t++)
                a[j][i]=t;
        for(int i=1;i<n-1;i+=2)
            a[i][k]^=a[i+1][k-1]^=a[i][k]^=a[i+1][k-1];
        if(k%2==0){
            for(int i=1;i<=n;i++,t++)
                a[n][i]=t;
            for(int i=1;i<n;i++,t++)
                a[i][n]=t;
        }
        else{
            ++t;
            for(int i=1;i<=n;i++,t++)
                a[n][i]=t;
            for(int i=1;i<n-1;i++,t++)
                a[i][n]=t;
            a[n-1][n]=(n-1)*(n-1)+1;
        }
    }
    for(int i=1;i<=n;i++,cout<<'\n')
        for(int j=1;j<=n;j++,cout<<' ')
            cout<<a[i][j];
}