快捷導(dǎo)航

python 實(shí)現(xiàn)矩陣按對(duì)角線(xiàn)打印

更新時(shí)間：2019年11月29日 09:26:19 作者：iamaline

今天小編就為大家分享一篇python 實(shí)現(xiàn)矩陣按對(duì)角線(xiàn)打印，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

如下所示：

Description：
將一個(gè)矩陣（二維數(shù)組）按對(duì)角線(xiàn)向右進(jìn)行打印。（搜了一下發(fā)現(xiàn)好像是美團(tuán)某次面試要求半小時(shí)手撕的題）
Example:
Input:
[
[1,2,3,4],
[5,1,2,3],
[9,5,1,2]
]
Output:
[[4],
[3, 3],
[2, 2, 2],
[1, 1, 1],
[5, 5],
[9]]

思路： 考慮每條對(duì)角線(xiàn)開(kāi)頭元素的index(i,j)。i從0開(kāi)始遍歷，j從col-1開(kāi)始遍歷，首先考慮j的變化，若j變?yōu)?，則保持不變，讓i變化。在確定開(kāi)頭元素后，可通過(guò)設(shè)置i+1，j+1及不超過(guò)范圍來(lái)向lst中對(duì)角線(xiàn)上添加元素。

對(duì)于以上例子，各對(duì)角線(xiàn)開(kāi)頭元素的index分別為：

(0,3)
(0,2)
(0,1)
(0,0)
(1,0)
(2,0)

其中各lst中的元素index分別為：

[(0,3)]
[(0,2),(1,3)]
[(0,1),(1,2),(2,3)]
[(0,0),(1,1),(2,2)]
[(1,0),(2,1)]
[(2,0)]

代碼：

def diagonal_right(matrix):
 if not matrix:
  return []
 row = len(matrix)
 col = len(matrix[0])
 col2 = col
 result = []
 for i in range(row):
  for j in range(col2 - 1, -1, -1): #j倒序遍歷
   lst = []
   i1,j1 = i,j #i1,j1用于方便同一對(duì)角線(xiàn)元素的添加，否則改變i,j影響開(kāi)頭元素的選擇
   while i1 <= row - 1 and j1 <= col - 1:
    lst.append(matrix[i1][j1])
    j1 += 1
    i1 += 1
   result.append(lst)
   if i == 0 and j == 0:#當(dāng)遍歷完(0,0)開(kāi)頭的一條對(duì)角線(xiàn)后，讓j固定為0
    col2 = 1
 return('the result is: %s'%result,'end')
 
#test
print(diagonal_right([[1],[2]]))
print(diagonal_right([[1]]))
print(diagonal_right([
 [1,2,3,4],
 [5,1,2,3],
 [9,5,1,2]
]))

輸出：

變型：由(0,0)開(kāi)始向左打印：

思路： 同樣考慮每條對(duì)角線(xiàn)開(kāi)頭元素的index(i,j)。i從0開(kāi)始遍歷，j也從0開(kāi)始遍歷，首先考慮j的變化，若j變?yōu)閏ol-1，則保持不變，讓i變化。在找到開(kāi)頭元素后，可通過(guò)設(shè)置i+1，j-1及不超過(guò)范圍來(lái)向lst中對(duì)角線(xiàn)上添加元素。

對(duì)于以上例子，各對(duì)角線(xiàn)開(kāi)頭元素的index分別為：

(0,0)
(0,1)
(0,2)
(0,3)
(1,3)
(2,3)

其中各lst中的元素index分別為：

[(0,0)]
[(0,1),(1,0)]
[(0,2),(1,1),(2,0)]
[(0,3),(1,2),(2,1)]
[(1,3),(2,2)]
[(2,3)]

代碼：

def diagonal_left(matrix):
 if not matrix:
  return []
 row = len(matrix)
 col = len(matrix[0])
 k=0
 result = []
 for i in range(row):
  for j in range(k,col): # j順序遍歷
   lst = []
   i1, j1 = i, j # i1,j1用于方便同一對(duì)角線(xiàn)元素的添加，否則改變i,j影響開(kāi)頭元素的選擇
   while i1 <= row - 1 and j1 >=0:
    lst.append(matrix[i1][j1])
    j1 -= 1
    i1 += 1
   if i==0 and j==col-1:
    k=col-1
   result.append(lst)
 return ('the result is: %s' % result, 'end')

# test
print(diagonal_left([[1],[2]]))
print(diagonal_left([[1]]))
print(diagonal_left([
 [1,2,3,4],
 [5,1,2,3],
 [9,5,1,2]
]))

輸出：