数据结构与算法刷题笔记——第一周3：一维数组存储二维矩阵

题目
助教小卢喜欢一种NN的矩阵：它由数字 1,2,3…NN 构成，矩阵中的元素各不相同，且每行、每列及矩阵的两条对角线上的数字之和都相同。
助教小卢称这种矩阵为优美矩阵，并且在N为奇数时，因为助教小卢是一个优雅的人，可以通过优雅的方式生成这一矩阵。

首先将 1 写在第一行的中间。
之后，按如下方式从小到大依次填写每个数 (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 的优雅矩阵。
进阶：使用一维数组存储这个矩阵。

例1：
输入：N = 5
输出：
17 24 1 8 15
23 5 7 14 16
4 6 13 20 22
10 12 19 21 3
11 18 25 2 9

请提交如下形式的函数：
void make_matrix(int N) {

}

这道题并不难，关键在于用一维数组存储二维矩阵，假设有一个矩阵（a_ij）_n*m（0<=i<n,0<=j<m）,index为矩阵中某个元素在一维数组的索引,那么我们只需利用公式 index = i * m + j 即可完成从二维矩阵到一维数组的映射。

具体代码如下：

#include <stdio.h>
#include <stdlib.h>

void make_matrix(int N) ;

int main()
{
    int N;
    scanf("%d",&N);
    make_matrix(N);
}

void make_matrix(int N) {
    int i,j,index,flag;
    int a[1000] = {0};
    flag = N/2;
    a[flag] = 1;
    int k;
    for(k = 2;k <= N*N;k++)
    {
        for(index = 0;index < N*N;index++)//index = i * N + j
        {
            i = index / N;//用一维数组保存二维矩阵
            j = index % N;
            if(a[index] == k - 1)
            {
                if(i == 0 && j!= N - 1)
                {
                    flag = (N-1) * N + j + 1;
                    a[flag] = k;
                }
                else if(j == N - 1 && i != 0)
                {
                    flag = (i - 1) * N + 0;
                    a[flag] = k;
                }
                else if(i == 0 && j == N - 1)
                {
                    flag = (i+1) * N +j;
                    a[flag] = k;
                }
                else if(i != 0 && j != N - 1)
                {
                    flag = (i - 1) * N + j + 1;
                    if(a[flag] == 0)
                    {
                        a[flag] = k;
                    }
                    else
                    {
                        flag = (i+1) * N + j;
                        a[flag] = k;
                    }
                }
            }
        }
    }
    for(index = 0;index < N*N;index++)
    {
        printf("%d ",a[index]);
    }
}