結果

問題 No.217 魔方陣を作ろう
コンテスト
ユーザー t98slider
提出日時 2022-07-31 18:27:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 1,763 bytes
コンパイル時間 2,364 ms
コンパイル使用メモリ 181,300 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 23:57:33
合計ジャッジ時間 3,628 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 14 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int N;

int safe_mod(int v){
    v %= N;
    if(v < 0)v += N;
    return v;
}

int main(){
    cin >> N;
    vector<vector<int>> A(N, vector<int>(N));
    int cnt = 1;
    if(N & 1){
        for(int i = 0; i < N; i++){
            for(int j = 0; j < N; j++){
                A[safe_mod(2 * i - j)][safe_mod(N / 2 - i + j)] = cnt++;
            }
        }
    }else if(N % 4 == 0){
        vector<vector<bool>> B = {{1,0,0,1},{0,1,1,0},{0,1,1,0},{1,0,0,1}};
        for(int i = 0; i < N; i++){
            for(int j = 0; j < N; j++){
                A[i][j] = (B[i][j] ? i * N + j + 1: N * N - (i * N + j));
            }
        }
    }else{
        N /= 2;
        vector<vector<int>> B(N, vector<int>(N));
        cnt--;
        for(int i = 0; i < N; i++){
            for(int j = 0; j < N; j++){
                B[safe_mod(2 * i - j)][safe_mod(N / 2 - i + j)] = 4 * cnt++;
            }
        }
        N *= 2;
        vector<vector<vector<int>>> L = {{{4, 1}, {2, 3}}, {{1, 4}, {2, 3}}, {{1, 4}, {3, 2}}};
        int c;
        for(int i = 0; i < N / 2; i++){
            for(int j = 0; j < N / 2; j++){
                if(i == N / 4 && j == N / 4)c = 1;
                else if(i == N / 4 + 1 && j == N / 4)c = 0;
                else c = (i <= N / 4 ? 0 : (i == N / 4 + 1 ? 1 : 2));
                for(int k = 0; k < 2; k++){
                    for(int l = 0; l < 2; l++){
                        A[2 * i + k][2 * j + l] = B[i][j] + L[c][k][l];
                    }
                }
            }
        }
    }

    for(int i = 0; i < N; i++){
        for(int j = 0; j < N; j++){
            if(j)cout << " ";
            cout << A[i][j];
        }
        cout << '\n';
    }
}
0