python 如何确定list每一列的高度是多少

Question

现在有一个list （每个字符串都是" . "或 “X”）。,如：

[['.','.','.','.','.','.','X','X'],
 ['.','.','.','.','.','X','X','X'],
 ['.','.','.','.','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','X','X','X','X','X','X'],
 ['X','X','X','X','X','X','X','X']]

输入一个阈值代表X的高度，然后返回小于等于阈值高度的列求和，

如果输入5，返回9  （也就是高度小于等于5的列全部求和）
输入7，返回22 

问题是想让代码的时间复杂度为O(n),

谢谢

Answer 1

不考虑复杂度的话直接遍历就是了。

A=[['.','.','.','.','.','.','X','X'],
 ['.','.','.','.','.','X','X','X'],
 ['.','.','.','.','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','.','X','X','X','X','X'],
 ['.','.','X','X','X','X','X','X'],
 ['X','X','X','X','X','X','X','X']]

def test(l,n):
    row, col = len(l), len(l[0])
    res = [0] * (row + 1)
    for i in range(col):
        cols=[rows[i] for rows in l]
        h=cols.count("X")
        res[h]+=h
    return sum(res[:n+1])
print(test(A,5))