結果

問題 No.217 魔方陣を作ろう
ユーザー fura
提出日時 2020-07-25 02:16:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,287 bytes
コンパイル時間 2,257 ms
コンパイル使用メモリ 199,492 KB
最終ジャッジ日時 2025-01-12 05:13:21
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:23:21: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   23 |         int n; scanf("%d",&n);
      |                ~~~~~^~~~~~~~~

ソースコード

diff # 

#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;
}
0