B 問題の diff 茶に初チャレンジ。
試験当日で、このレベルが来ると自分なら焦ると思います。
Candles.py
H,W = map(int,input().split())
A = [list(input()) for _ in range(H)]
lisA = []
h = 0
for h in range(H):
if A[h].count(".") != W:
lisA.append(A[h])
#print(lisA,len(lisA))
memo = []
for w in range(W):
cnt = 0
for h in range(len(lisA)):
#print(lisA[h][w],h,w)
if lisA[h][w] == ".":
cnt += 1
if cnt == len(lisA):
memo.append(w)
memo = set(memo)# <= ココ
ans =[]
for h in range(len(lisA)):
ans.append([])
for w in range(W):
if w not in memo:# <= ココ
ans[h].append(lisA[h][w])
#print(ans)
for i in range(len(ans)):
print("".join(ans[i]))
in 演算子は set は計算量 O(1)。コメントの**"ココ"** をご覧ください。