CodeSignal 8번 matrixElementsSum문제풀이

처음으로 만났던 어려운 문제인 7번이 지나가고 한동안은 쉬운 문제만 나옵니다.

 

이 문제는 2차원 배열 형태의 입력이 주어지고,  유령의 영향을 받지않은 방의 수만 더하는 문제입니다.

입력 행렬에서 0인방은 유령방이고, 그 아래 방들은 모두 숫자를 더하지 않습니다.

이 문제는 다음과 같이 풀었습니다.

def matrixElementsSum(matrix):
    ghostLine=[]
    sum=0
    for floor in matrix:
        for line in range(len(matrix[0])):
            if line in ghostLine: continue
            if floor[line]==0: ghostLine.append(line)
            sum+=floor[line]
    return sum

 

유령방이 있는 열들을 ghostLine에 저장하며 푸는 방식입니다.