【題目】
給定一個整型矩陣 matrix,請按照轉圈的方式列印它。
例如:
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
要求額外空間複雜度為O(1).
【基本思路】
一層一層的列印矩陣。在矩陣中用左上角(tR, tC)和右下角(dR, dC)就可以表示一個子矩陣。比如,題目中的矩陣,當(tR, tC) = (0,0), (dR, dC) = (3,3)。表示的子矩陣就是整個矩陣,它的最外圈的部分如下:
1 2 3 4
5 8
9 12
13 14 15 16
轉圈把這個外層列印出來,然後令tR, tC加1,dR, dC減1,這時所表示的子矩陣就是:
6 7
10 11
繼續轉圈列印出來即可。
def spiralOrderPrint(arr):
tr = tc = 0
#(tr,tc)為數組左上角的坐标,開始0行0列
dr = len(arr)-1
#dr 表示右下角行的坐标索引,len(arr)代表數組行數 -1為索引
dc = len(arr[0])-1
#dc 表示右下角列的索引 len(arr[0])表示每行有幾個元素 -1為索引
while tc<=dc and tr<=dr:
#條件確定左列不超過右列。確定下行不超過上行
if tr == dr:
for i in range(tc,dc+1):
print(arr[tr][i],end=' ')
elif tc==dc:
for i in range(tr,dr+1):
print(arr[i][tc],end=' ')
else:
for i in range(tc,dc):
print(arr[tr][i],end=' ')
for i in range(tr,dr):
print(arr[i][dc],end=' ')
for i in range(dc,tc,-1):
print(arr[dr][i],end=' ')
for i in range(dr,tr,-1):
print(arr[i][tc],end=' ')
tc+=1
tr+=1
dc-=1
dr-=1
if __name__ == '__main__':
li = [[1,2,3,4],
[5,6,7,8],
[9,10,11,12],
[13,14,15,16]]
spiralOrderPrint(li)