Leetcode#59 Spiral Matrix II

原题地址

相比于Spiral Matrix（参见这篇文章）要简单一些，因为是方阵，所以代码简洁一些。

注意当n是奇数的时候，中心小块要单独赋值（代码21行）

代码：

 vector<vector<int> > generateMatrix(int n) {
   vector<vector<int> > matrix(n, vector<int>(n, ));
   int len = n - ;
   int i = ;
   int j = ;
   int count = ;

   while (len > ) {
     for (int k = ; k < len; k++)
       matrix[i][j++] = count++;
     for (int k = ; k < len; k++)
       matrix[i++][j] = count++;
     for (int k = ; k < len; k++)
       matrix[i][j--] = count++;
     for (int k = ; k < len; k++)
       matrix[i--][j] = count++;
     i++;
     j++;
     len -= ;
   }
   if (len == )
     matrix[i][j] = count;

   return matrix;
 }