P2615 [NOIP2015 提高组] 神奇的幻方

题目链接：https://www.luogu.com.cn/problem/P2615

题目描述

幻方是一种很神奇的 N*N 矩阵：它由数字 1,2,3,… ,N×N 构成，且每行、每列及两条对角线上的数字之和都相同。

当 N 为奇数时，我们可以通过下方法构建一个幻方：

首先将 1 写在第一行的中间。

之后，按如下方式从小到大依次填写每个数 K (K=2,3,⋯,N×N) ：

• 若 (K-1) 在第一行但不在最后一列，则将 K 填在最后一行， (K-1) 所在列的右一列；
• 若 (K-1) 在最后一列但不在第一行，则将 K填在第一列， (K-1) 所在行的上一行；
• 若 (K-1)在第一行最后一列，则将 K 填在 (K-1) 的正下方；
• 若 (K-1) 既不在第一行，也不在最后一列，如果 (K-1) 的右上方还未填数，则将 K 填在(K−1) 的右上方，否则将 K 填在(K−1) 的正下方。

现给定 N ，请按上述方法构造 N×N 的幻方。

输入格式

一个正整数 N ，即幻方的大小。

输出格式

共 N 行 ，每行 N 个整数，即按上述方法构造出的 N×N 的幻方，相邻两个整数之间用单空格隔开。

输入输出样例

3

8 1 6
3 5 7
4 9 2

题目分析

使用一个二维数组来存这个幻方。从1开始，依次按照构造规则进行判断是属于哪一种情况，将数字填入指定位置，并记录刚填写的 x、y 坐标

代码实现

#include <iostream>
using namespace std;
int n, g[40][40], x, y;

int main() {
    cin >> n;
    g[1][n / 2 + 1] = 1;
    x = 1;
    y = n / 2 + 1;
    for (int i = 2; i <= n * n; i++) {
        if (x == 1 && y != n) {     // 在第一行但不在最后一列
            g[n][y + 1] = i;
            x = n;
            y++;
        }
        else if (y == n && x != 1) {    // 最后一列但不是第一行 
            g[x-1][1] = i;
            x--;
            y = 1;
        }
        else if (x == 1 && y == n) {   // 第一行最后一列
            g[2][n] = i;
            x = 2;
        }
        else if (x != 1 && y != n) {   // 不在第一行，也不再最后一列
            if (g[x-1][y+1] == 0) {
                g[x-1][y+1] = i;
                x--;
                y++;
            }
            else {
                g[x+1][y] = i;
                x++;
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) 
            cout << g[i][j] << " ";
        cout << endl;
    }
    return 0;
}