现在有一个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个数相加?