洛谷 #2615. 神奇的幻方

题意

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

若 (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) 的正下方。

输出构造的幻方（废话）

题解

模拟即可，保留上一个的x，y坐标

调试记录

无

#include <cstdio>
#include <cstring>

using namespace std;

long long maze[41][41];

int main(){
    memset(maze, 0, sizeof maze);
    int n; scanf("%d", &n);
    
    int now = 1, prex, prey;
    maze[prex = 1][prey = n / 2 + 1] = now++;
    
    while (now <= n * n){
        bool flag1, flag2;
        flag1 = (prex == 1);
        flag2 = (prey == n);
        
        if (flag1 && !flag2) maze[prex = n][++prey] = now;
        if (!flag1 && flag2) maze[--prex][prey = 1] = now;
        if (flag1 && flag2) maze[++prex][prey] = now;
        if (!flag1 && !flag2)
            if (maze[prex - 1][prey + 1] == 0) maze[--prex][++prey] = now;
            else maze[++prex][prey] = now;
        now++;
    }
    
    for (int i = 1; i <= n; i++){ for (int j = 1; j < n; j++) printf("%d ", maze[i][j]); printf("%d\n", maze[i][n]); }
    
    return 0;
}