48. 旋转图像

59 阅读1分钟

题目

给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。

你必须在 原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像题目链接

分析

数学找规律题,旋转之后 matrix[i][j] = m[n-j-i][i]。旋转之后四个元素为一个循环,所以计算四次。 注意两点

  1. 循环跳出条件,注意 n 为奇数和偶数时截止条件分别是什么,需保证每个元素只计算一次
  2. 优化之处在于将一个循环中四个元素旋转后的位置全部用i, j计算表达出来 以代替一个循环中不断更换i,j实时计算,这种优化对计算机而言影响甚微,没必要推算
fun rotate(matrix: Array<IntArray>): Unit {
    val n = matrix.size
    for (i in 0 until (n + 1) / 2) {
        for (j in 0 until n / 2) {
            var row = i
            var col = j
            //一个循环四个元素 依次计算
            var temp = matrix[i][j]
            for (i in 0..2) {
                val nextRow = n - col - 1
                val nextCol = row
                matrix[row][col] = matrix[nextRow][nextCol]
                row = nextRow
                col = nextCol
            }
            matrix[row][col] = temp
        }
    }
}

关于我

一个希望友友们能提出建议的代码下毒糕手