2016校招真题之顺时针旋转矩阵

1、题目描述

有一个NxN整数矩阵，请编写一个算法，将矩阵顺时针旋转90度。给定一个NxN的矩阵，和矩阵的阶数N,请返回旋转后的NxN矩阵,保证N小于等于300。

测试样例：

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

返回：

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

 

2、代码实现

package com.wcy.october;

/**
 * 时间：2016年10月16日
 * 题目：顺时针旋转矩阵
 * 题目描述：有一个NxN整数矩阵，请编写一个算法，将矩阵顺时针旋转90度。给定一个NxN的矩阵，和矩阵的阶数N,请返回旋转后的NxN矩阵,保证N小于等于300。
 * 测试样例：[[1,2,3],[4,5,6],[7,8,9]],3 返回：[[7,4,1],[8,5,2],[9,6,3]]
 */
public class Rotate {

    /**
     * 方法一：将矩阵顺时针旋转90度
     * @param mat 需要顺时针旋转90度的矩阵
     * @param n 矩阵的大小
     * @return 顺时针旋转90度后的新矩阵
     */
    public int[][] rotateMatrix(int[][] mat, int n) {
        int[][] result = new int[n][n];
        for (int i = 0; i < mat.length; i++) {
            int num = 0;
            for (int j = mat.length-1; j >= 0&&num < mat.length; j--,num++) {
                result[i][num] = mat[j][i];
            }
        }
        return result;
    }
    
    /**
     * 方法二：将矩阵顺时针旋转90度，找规律：mat[i][j]被旋转到了mat[j][n-i-1]的位置
     * @param mat 需要顺时针旋转90度的矩阵
     * @param n 矩阵的大小
     * @return 顺时针旋转90度后的新矩阵
     */
    public int[][] rotateMatrix2(int[][] mat, int n) {
        int[][] result = new int[n][n];
        for (int i = 0; i < mat.length; i++) {
            for (int j = 0; j < mat.length; j++) {
                result[j][n-i-1] = mat[i][j];
            }
        }
        return result;
    }
    
    /**
     * 用户页面测试
     * @param args
     */
    public static void main(String[] args) {
        Rotate rotate = new Rotate();
        int[][] mat = {{1,2,3,5,6},{7,8,9,10,11},{12,13,14,15,16},{17,18,19,20,21},{22,23,24,25,26}};
        int n = mat.length;
        int[][] result = rotate.rotateMatrix(mat, n);
        // 打印出结果
        for (int i = 0; i < result.length; i++) {
            for (int j = 0; j < result.length; j++) {
                if (j != result.length-1) {
                    System.out.print(result[i][j] + " ");
                }else {
                    System.out.print(result[i][j]);
                }
            }
            System.out.println();
        }
    }
}