generateMatrix 给定N 返回指定矩阵
2016-06-14 12:19
369 查看
给定N= 3,返回如下数组1 2 38 9 47 6 5
算法: 每次从 (0,0) (1,1) (2,2) 开始,共开始n/2次。每次矩阵规模缩小2*2代码 class Solution {
算法: 每次从 (0,0) (1,1) (2,2) 开始,共开始n/2次。每次矩阵规模缩小2*2代码 class Solution {
public: vector<vector<int>> generateMatrix(int n) { vector<vector<int>> result; unsigned long m = (unsigned long)n; result.resize(m); for (int i1 = 0; i1 < m; ++i1) { result[i1].resize(m); } int k = n / 2 ; if(n%2 == 0) k-=1; int count = 0; int begin = 1; for (int i = 0; i <= k; ++i) { for (int j = i; j < n; ++j) { result[i][j] = begin++; } for (int l = i + 1; l < n; ++l) { result[l][n-1] = begin ++; } for (int s = n-2; s >=i ; s--) { result[n-1][s] = begin ++; } for (int p = n-2; p > i ; p--) { result[p][i] = begin ++; } n -= 1; } return result; } };