LeetCode之螺旋矩阵

Posted 2020-11-18 youdias

问题

螺旋矩阵
给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 1:

输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]]
输出: [1,2,3,6,9,8,7,4,5]
示例 2:

输入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

分析

限定好前进方向以及边界，一个一个遍历下去即可，最近开始用Go，感觉很是不错，特意选了Go练习，AC code如下
···

func spiralOrder(matrix [][]int) []int {
var ret []int
ret = make([]int, 0)
n := len(matrix)
if n == 0 {
return ret
}
m := len(matrix[0])

//获取到了矩阵大小，
counter := 0 //记录处理个数
limu, limd, liml, limr := -1, n, -1, m
tmpx, tmpy := 0, 0
direction := 0 // 0 right 1 down 2 left7 3 up
for {
    if counter == n*m {
        return ret
    }
    ret = append(ret, matrix[tmpx][tmpy])
    counter++
    getNextIndex(&limu, &limd, &liml, &limr, &direction, &tmpx, &tmpy)
}

}

//获取下一个坐标，同时有可能更新方向、边界等
func getNextIndex(limu int, limd int, liml int, limr int, direction int, x int, y int) {
//fmt.Println(limu, limd, liml, limr, direction, x, y)
switch direction {
case 0:
//to right
if y+1 < limr {
//正常更新
y += 1
return
} else {
//到了右边界，需要更新方向以及边界，已经通过个数限定，所以改变方向后肯定可行
limu += 1
direction = 1
x += 1
}
case 1:
//to down
if x+1 < limd {
//正常更新
x += 1
return
} else {
//到了下边界，需要更新方向以及边界，
limr -= 1
direction = 2
y -= 1
}
case 2:
//to right
if y-1 > liml {
//正常更新
y -= 1
return
} else {
//到了左边界，需要更新方向以及边界，
limd -= 1
direction = 3
x -= 1
}
case 3:
//to up
if x-1 > limu {
//正常更新
x -= 1
return
} else {
//到了左边界，需要更新方向以及边界，已经通过个数限定，所以改变放下后肯定可行
liml += 1
direction = 0
*y += 1
}
}
}

···

其他

1.初次提交后由于空输入错了一次，应该更多的考虑特殊情况。
2.使用go的另一个好处就是可以很方便的进行测试，比如main.go中编写了解题函数，只需要一个公开的壳函数就能在mai_test.go中测试
···

func TspiralOrder(matrix [][]int) []int {
return spiralOrder(matrix)
}

···
而在测试中通过编写测试数据测试结果，可以方便的添加新的测试数据，同时能够避免改动导致原本能通过的数据failed。
···

func TestTspiralOrder(t *testing.T) {
var testData [][][]int
var tmpdata [][]int
var testReslut [][]int
var tmpanswer []int

tmpdata = [][]int{
    {1, 2, 3},
    {4, 5, 6},
    {7, 8, 9}}
testData = append(testData, tmpdata)
tmpanswer = []int{1, 2, 3, 6, 9, 8, 7, 4, 5}
testReslut = append(testReslut, tmpanswer)

tmpdata = [][]int{
    {1, 2, 3, 4},
    {5, 6, 7, 8},
    {9, 10, 11, 12}}
testData = append(testData, tmpdata)
tmpanswer = []int{1, 2, 3, 4, 8, 12, 11, 10, 9, 5, 6, 7}
testReslut = append(testReslut, tmpanswer)

tmpdata = make([][]int, 0)
testData = append(testData, tmpdata)
tmpanswer = make([]int, 0)
testReslut = append(testReslut, tmpanswer)

for i := 0; i != len(testData); i++ {
    tmp := TspiralOrder(testData[i])
    fmt.Println(tmp)
    if IsTwoArraySame(tmp, testReslut[i]) == true {
        fmt.Println("Pass")
    } else {
        fmt.Println("Failed")
    }
}

}
···