#518078 (C++17) No.217 魔方陣を作ろう

• 問題: No.217 魔方陣を作ろう
• ユーザー: fura
• 提出日時: 2020-07-25 02:16:46
• 言語: C++17 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 2 ms / 5,000 ms
• コード長: 1,287 bytes
• コンパイル時間: 2,806 ms
• コンパイル使用メモリ: 204,496 KB
• 実行使用メモリ: 4,384 KB
• 最終ジャッジ日時: 2023-09-08 09:51:13
• 合計ジャッジ時間: 3,507 ms
• ジャッジサーバーID （参考情報）: judge15 / judge14

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,380 KB
testcase_01	AC	2 ms; 4,380 KB
testcase_02	AC	2 ms; 4,380 KB
testcase_03	AC	1 ms; 4,380 KB
testcase_04	AC	1 ms; 4,380 KB
testcase_05	AC	1 ms; 4,380 KB
testcase_06	AC	2 ms; 4,384 KB
testcase_07	AC	2 ms; 4,380 KB
testcase_08	AC	1 ms; 4,384 KB
testcase_09	AC	1 ms; 4,376 KB
testcase_10	AC	2 ms; 4,380 KB
testcase_11	AC	2 ms; 4,380 KB
testcase_12	AC	2 ms; 4,376 KB
testcase_13	AC	2 ms; 4,384 KB
testcase_14	AC	2 ms; 4,380 KB
testcase_15	AC	2 ms; 4,380 KB
testcase_16	AC	2 ms; 4,380 KB
testcase_17	AC	1 ms; 4,380 KB

ソースコード

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;

vector<vector<int>> solve_odd(int n){
	vector B(n,vector(n,0));
	int i=0,j=n/2;
	rep(k,n*n){
		B[i][j]=k+1;
		i=(i-1+n)%n;
		j=(j+1)%n;
		if(B[i][j]!=0){
			i=(i+2)%n;
			j=(j-1+n)%n;
		}
	}
	return B;
}

int main(){
	int n; scanf("%d",&n);

	vector B(n,vector(n,0));
	if(n%2==1){
		B=solve_odd(n);
	}
	else if(n%4==0){
		rep(i,n) rep(j,n) {
			if(i%4==j%4 || i%4+j%4==3) B[i][j]=i*n+j+1;
		}
		for(int i=n-1;i>=0;i--) for(int j=n-1;j>=0;j--) {
			if(i%4!=j%4 && i%4+j%4!=3) B[i][j]=(n-i-1)*n+(n-j-1)+1;
		}
	}
	else{
		auto C=solve_odd(n/2);
		rep(i,n/2) rep(j,n/2) C[i][j]=4*(C[i][j]-1);
		rep(i,n/2) rep(j,n/2) {
			// L
			if(i<n/4 || (i==n/4 && j!=n/4) || (i==n/4+1 && j==n/4)){
				B[2*i+0][2*j+0]=C[i][j]+4;
				B[2*i+0][2*j+1]=C[i][j]+1;
				B[2*i+1][2*j+0]=C[i][j]+2;
				B[2*i+1][2*j+1]=C[i][j]+3;
			}
			// U
			else if(i<=n/4+1){
				B[2*i+0][2*j+0]=C[i][j]+1;
				B[2*i+0][2*j+1]=C[i][j]+4;
				B[2*i+1][2*j+0]=C[i][j]+2;
				B[2*i+1][2*j+1]=C[i][j]+3;
			}
			// X
			else{
				B[2*i+0][2*j+0]=C[i][j]+1;
				B[2*i+0][2*j+1]=C[i][j]+4;
				B[2*i+1][2*j+0]=C[i][j]+3;
				B[2*i+1][2*j+1]=C[i][j]+2;
			}
		}
	}

	rep(i,n) rep(j,n) printf("%d%c",B[i][j],j<n-1?' ':'\n');

	return 0;
}