剑指offer 顺时针打印矩阵

Posted on |

顺时针打印矩阵

题目

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下4 X 4矩阵： 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.

方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# -*- coding:utf-8 -*-
class Solution:
    # matrix类型为二javascript:void(0);维列表，需要返回列表
    def printMatrix(self, matrix):
        # write code here
        def coord(r1,r2,c1,c2):
            for c in range(c1,c2+1):
                yield r1,c
            for r in range(r1+1,r2+1):
                yield r,c2
            if r1<r2 and c1<c2:
                for c in range(c2-1,c1,-1):
                    yield r2,c
                for r in range(r2,r1,-1):
                    yield r,c1
        if not matrix:
            return []
        ans=[]
        r1,r2=0,len(matrix)-1
        c1,c2=0,len(matrix[0])-1
        while r1<=r2 and c1<=c2:
            for r,c in coord(r1,r2,c1,c2):
                ans.append(matrix[r][c])
            r1+=1;r2-=1
            c1+=1;c2-=1
        return ans