NOIP2015提高组 神奇的幻方

NOIP2015提高组 神奇的幻方

题目描述

幻方是一种很神奇的N*N矩阵：它由数字1,2,3,……,N*N构成，且每行、每列及两条对角线上的数字之和都相同。
当N为奇数时，我们可以通过以下方法构建一个幻方：
首先将1写在第一行的中间。
之后，按如下方式从小到大依次填写每个数K(K=2,3,…,N*N)：
1.若(K−1)在第一行但不在最后一列，则将K填在最后一行，(K−1)所在列的右一列；
2.若(K−1)在最后一列但不在第一行，则将K填在第一列，(K−1)所在行的上一行；
3.若(K−1)在第一行最后一列，则将K填在(K−1)的正下方；
4.若(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
说明
对于100%的数据，1<=N<=39且N为奇数
来源：NOIp2015 提高组 d1t1

分析：根据题意判断下一个数存放的位置。需另开一个数组存放某个位置是否已填充。

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           using namespace std; #define N 40 int n, magic[N][N], filled[N][N] = {0}; struct node{ int row; int col; void set(int r, int c, int v){ row = r; col = c; magic[r][c] = v; filled[r][c] = 1; } }pos[N*N]; int read_i(){ char ch = getchar(); int ans = 0; while(ch>'9' || ch<'0') ch = getchar(); while(ch<='9' && ch>='0'){ ans = ans*10 + ch -'0'; ch = getchar(); } return ans; } int main(){ ios::sync_with_stdio(false); int k = 1, row, col, i, j; n = read_i(); pos[k].set(0, n/2, k); for(k=2; k<=n*n; k++){ row = pos[k-1].row, col = pos[k-1].col; if(row == 0 && col != n-1) pos[k].set(n-1, col+1, k); else if(row != 0 && col == n-1) pos[k].set(row-1, 0, k); else if(row == 0 && col == n-1) pos[k].set(row+1, col, k); else if(row != 0 && col != n-1){ if(filled[row-1][col+1] == 0) pos[k].set(row-1, col+1, k); else pos[k].set(row+1, col, k); } } for(i=0; i