Code Signal_练习题_matrixElementsSum

Posted 2020-12-24 yd2018

After they became famous, the CodeBots all decided to move to a new building and live together. The building is represented by a rectangular matrix of rooms. Each cell in the matrix contains an integer that represents the price of the room. Some rooms are free(their cost is 0), but that‘s probably because they are haunted, so all the bots are afraid of them. That is why any room that is free or is located anywhere below a free room in the same column is not considered suitable for the bots to live in.

Help the bots calculate the total price of all the rooms that are suitable for them.

Example

• For

matrix = [[0, 1, 1, 2], 
          [0, 5, 0, 0], 
          [2, 0, 3, 3]]

the output should be
matrixElementsSum(matrix) = 9.

Here‘s the rooms matrix with unsuitable rooms marked with ‘x‘:

[[x, 1, 1, 2], 
 [x, 5, x, x], 
 [x, x, x, x]]

Thus, the answer is 1 + 5 + 1 + 2 = 9.

• For

matrix = [[1, 1, 1, 0], 
          [0, 5, 0, 1], 
          [2, 1, 3, 10]]

the output should be
matrixElementsSum(matrix) = 9.

Here‘s the rooms matrix with unsuitable rooms marked with ‘x‘:

[[1, 1, 1, x], 
 [x, 5, x, x], 
 [x, 1, x, x]]

Note that the free room in the first row make the full column unsuitable for bots.

Thus, the answer is 1 + 1 + 1 + 5 + 1 = 9.

 

我的解答：

 1 def matrixElementsSum(matrix):
 2     num = 0
 3     j = 0
 4     while j<len(matrix):
 5         num = num + sum(matrix[j])
 6         for i in range(len(matrix[j])):
 7             if matrix[j][i] == 0:
 8                 for x in range(j+1,len(matrix)):
 9                     matrix[x][i] = 0
10         j += 1
11     return num

 

膜拜大佬：

 1 def matrixElementsSum(m):
 2     r = len(m)
 3     c = len(m[0])
 4     total=0
 5     for j in range(c):
 6         for i in range(r):
 7             if m[i][j]!=0:
 8                 total+=m[i][j]
 9             else:
10                 break
11     return total

View Code