Leetcode#59 Spiral Matrix II

原题地址

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

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

代码：

vector<vector<int> > generateMatrix(int n) {
  vector<vector<int> > matrix(n, vector<int>(n, 0));
  int len = n - 1;
  int i = 0;
  int j = 0;
  int count = 1;
        
  while (len > 0) {
    for (int k = 0; k < len; k++)
      matrix[i][j++] = count++;
    for (int k = 0; k < len; k++)
      matrix[i++][j] = count++;
    for (int k = 0; k < len; k++)
      matrix[i][j--] = count++;
    for (int k = 0; k < len; k++)
      matrix[i--][j] = count++;
    i++;
    j++;
    len -= 2;
  }
  if (len == 0)
    matrix[i][j] = count;
            
  return matrix;
}