懒人读算法（七）-旋转矩阵

趣味题

给一个数字n，你要画出从1到n平方的旋转矩阵
如n=3，则你要画出一下的一个数组
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]

答案:

public class Solution {

    public int[][] generateMatrix(int n) {


            int[][] spiral = new int[n][n];

            if(n == 0) return spiral;

            int rowStart = 0;
            int colStart = 0;
            int rowEnd = n - 1;
            int colEnd = n -1;
            int number = 1;

            while(rowStart <= rowEnd && colStart <= colEnd) {

                for(int i = colStart; i <= colEnd; i++) {//列开始->列结束

                    spiral[rowStart][i] = number++;//行确定，逐列增加

                }//确定1,2,3

                rowStart++;//行开始加1

                for(int i = rowStart; i <= rowEnd; i++) {//行开始->行结束

                    spiral[i][colEnd] = number++;//最后一列的行,逐行增加

                }//确定4,5

                colEnd--;//列结束-1   也就是列结束要前移了

                for(int i = colEnd; i >= colStart; i--) {//列结束->列开始

                    if(rowStart <= rowEnd) {

                        spiral[rowEnd][i] = number++;

                    }

                }//确定6,7

                rowEnd--;//行结束-1,也就是行结束上移了

                for(int i = rowEnd; i >= rowStart; i--) {//行结束->行开始          在本例中这两个相等

                    if(colStart <= colEnd) {//列开始此时为0，列结束此时为1

                        spiral[i][colStart] = number++;

                    }

                }//确定8,9

                colStart++;//列开始增加1

            }

            return spiral;


    }

}

解答:代码还是容易看懂得。