ABC107 B - Candles diff 차는 초조할지도

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). 댓글의 "코코"를 참조하십시오.