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

现在有一个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),

谢谢

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

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))

 

没看懂你的列求和是什么意思。列元素相加来时列数量相加还是x个数相加?