2020-04-11 6c7860d1e1e6a82aa7ae0079596b3f0d 99+ 2 分钟 0.2 k 0次访问

剑指offer-顺时针打印矩阵

一、题意

给定一个包含 n x m 个元素的矩阵（n 行，m 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。
例如，如果输入如下矩阵：
输入:
[
　　[1, 2, 3, 4],
　　[5, 6, 7, 8],
　　[9,10,11,12]
]
输出: [1, 2, 3, 4, 8, 12, 11, 10, 9, 5, 6, 7]

二、参考Leetcode原题

LeetCode 54. Spiral Matrix

三、代码实现

package main

import "fmt"

func PrintMatrix(matrix [][]int) []int {
	var res []int
	// 行数
	n := len(matrix)
	if n == 0 {
		return res
	}
	// 列数
	m := len(matrix[0])
	// 计算元素个数
	nm := n * m
	// 初始化层
	layer := 0
	// 初始化边界
	startN, endN, startM, endM := 0, 0, 0, 0
	// 从外围到内部顺时针遍历
	for nm > 0 {
		startN, endN = layer, n-layer-1
		startM, endM = layer, m-layer-1

		// 从左到右（行号不变）
		for i := startM; i <= endM && nm > 0; i++ {
			res = append(res, matrix[startN][i])
			nm--
		}

		// 从上到下（列号不变）
		for i := startN+1; i <= endN && nm > 0; i++ {
			res = append(res, matrix[i][endM])
			nm--
		}

		// 从右到左（行号不变）
		for i := endM-1; i >= startM && nm > 0; i-- {
			res = append(res, matrix[endN][i])
			nm--
		}

		// 从下到上（列号不变）
		for i := endN-1; i >= startN+1 && nm > 0; i-- {
			res = append(res, matrix[i][startM])
			nm--
		}

		// 层数递增
		layer++
	}
	return res
}

func main()  {
	var matrix [][]int = [][]int {{1, 2}, {4, 5}, {7, 8}}
	res := PrintMatrix(matrix)
	fmt.Print(res)
}

// 输出如下
[1 2 5 8 7 4]
Process finished with exit code 0