CodeSignal 刷題 —— matrixElementSum

標籤：return   tran   矩形   .com   決定   sse   group   elements   包含   

 

 題目意思：

當他們成名後，所有的代碼機器人都決定搬到新的建築裡住在一起。建築是用房間的矩形矩陣表示的。矩陣中的每個儲存格都包含一個表示房間價格的整數。有些房間是免費的(費用為0)，但這可能是因為它們是鬧鬼的，所以所有的機器人都害怕它們。這就是為什麼任何閒置房間或者位於同一列的空閑房間下面的任何地方都不適合機器人居住的原因。

 

 

def matrixElementsSum(matrix):    sum = 0    for i in range(len(matrix[0])):      # 先處理第一行        if matrix[0][i] != 0:            sum += matrix[0][i]    for i in range(1, len(matrix)):        for j in range(len(matrix[0])):            if matrix[i-1][j] == 0:    # 同一列中，如果上一行的元素為0，則把此處的元素也置為0                matrix[i][j] =0            sum += matrix[i][j]                return sum

 

CodeSignal 刷題 —— matrixElementSum