LeetCode54/59 Spiral Matrix I/II

来源:互联网 发布:mac显示隐藏文件命令 编辑:程序博客网 时间:2024/06/01 10:41

一:Spiral Matrix I

题目:

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 7, 8, 9 ]]

You should return [1,2,3,6,9,8,7,4,5].

链接:https://leetcode.com/problems/spiral-matrix/

分析:看代码,四个while循环形成一个大圈

class Solution {public:    void dfs(int xi, int yi, const int &m, const int &n, vector<vector<int> > &matrix, vector<int> &result, int **visited){        visited[xi][yi] = 1;        result.push_back(matrix[xi][yi]);        while(yi+1 < n && !visited[xi][yi+1]){  // 四个while循环刚好跑了大正圈            result.push_back(matrix[xi][yi+1]);            visited[xi][yi+1] = 1;            yi = yi+1;        }        while(xi+1 < m && !visited[xi+1][yi]){            result.push_back(matrix[xi+1][yi]);            visited[xi+1][yi] = 1;            xi = xi +1;        }        while(yi-1 >=0 && !visited[xi][yi-1]){            result.push_back(matrix[xi][yi-1]);            visited[xi][yi-1] = 1;            yi = yi-1;        }         while(xi-1 >=0 && !visited[xi-1][yi]){            result.push_back(matrix[xi-1][yi]);            visited[xi-1][yi] = 1;            xi = xi-1;        }        if(yi+1 < n && !visited[xi][yi+1]) dfs(xi, yi+1, m, n, matrix, result, visited);    }    vector<int> spiralOrder(vector<vector<int> > &matrix) {        vector<int> result;        int m = matrix.size();        if(m == 0) return result;                int n = matrix[0].size();        int **visited= new int*[m];        for(int i = 0; i < m; i++){            visited[i] = new int[n];            memset(visited[i], 0, sizeof(int)*n);        }        dfs(0, 0, m, n, matrix, result, visited);                for(int i = 0; i < m; i++)            delete [] visited[i];        delete []visited;        return result;                    }};

二:Spiral Matrix II

题目:

Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

For example,
Given n = 3,

You should return the following matrix:
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ]]
链接:https://leetcode.com/problems/spiral-matrix-ii/

class Solution {public:    void dfs(int xi, int yi, const int &n, vector<vector<int> > &result, int &value){        result[xi][yi] = value++;         while(yi+1 < n && !result[xi][yi+1]){  // 四个while循环刚好跑了大正圈            result[xi][yi+1] = value++;            yi = yi+1;        }        while(xi+1 < n && !result[xi+1][yi]){              result[xi+1][yi] = value++;            xi = xi+1;        }        while(yi-1 >= 0 && !result[xi][yi-1]){              result[xi][yi-1] = value++;            yi = yi-1;        }        while(xi-1 >= 0 && !result[xi-1][yi]){              result[xi-1][yi] = value++;            xi = xi-1;        }        if(yi+1 < n && !result[xi][yi+1]) dfs(xi, yi+1, n, result, value);    }        vector<vector<int> > generateMatrix(int n) {        vector<vector<int> >result;        if(n == 0) return result;        for(int i = 0; i < n; i++){  // 初始化result            vector<int> temp(n, 0);            result.push_back(temp);        }        int value = 1;        dfs(0,0, n, result, value);        return result;            }};


1 0
原创粉丝点击