牛客网: BM99
题目: 矩阵旋转90度
思路: 遍历交换, 对角线交换、左右交换
代码:
- // go
-
- package main
- // import "fmt"
-
- /**
- * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
- *
- *
- * @param mat int整型二维数组
- * @param n int整型
- * @return int整型二维数组
- */
- func rotateMatrix( mat [][]int , n int ) [][]int {
- // write code here
- for i := 0; i < n; i++ {
- for j := 0; j < i; j++ {
- mat[i][j], mat[j][i] = mat[j][i], mat[i][j]
- }
- }
- for j := 0; j < n/2; j++ {
- for i := 0; i < n; i++ {
- mat[i][j], mat[i][n-1-j] = mat[i][n-1-j], mat[i][j]
- }
- }
- return mat
- }