码迷,mamicode.com
首页 > 其他好文 > 详细

Code Signal_练习题_matrixElementsSum

时间:2018-07-22 00:14:44      阅读:229      评论:0      收藏:0      [点我收藏+]

标签:cost   div   span   ble   nal   color   present   elements   mat   

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

 

Code Signal_练习题_matrixElementsSum

标签:cost   div   span   ble   nal   color   present   elements   mat   

原文地址:https://www.cnblogs.com/YD2018/p/9348418.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!