[SOFTEER] 진정한 효도

문제 링크

https://softeer.ai/practice/7374

Softeer - 현대자동차그룹 SW인재확보플랫폼

느낀 점

파이썬 리스트의 이차원 배열에서 열을 가져오고 싶을 때, matrix[:][0]과 같은 형식으로 하면 안된다. 왜냐하면 [:]는 얕은 복사를 하기때문에 뒤에서 슬라이싱을하면 첫 번째 행을 가져옴 따라서 [m[0] for m in matrix] 와 같이 작성해야함

import sys
def judge(m):
    m.sort()
    if m == [1, 1, 1] or m == [2, 2, 2] or m == [3, 3, 3]:
        return 0
    elif m == [1, 1, 2] or m == [1, 2, 2] or m == [2, 2, 3] or m == [2, 3, 3]:
        return 1
    elif m == [1, 1, 3] or m == [1, 2, 3] or m == [1, 3, 3]:
        return 2
maps = []
answer = []
for _ in range(3):
    maps.append(list(map(int, sys.stdin.readline().split())))
for i in range(3):
    m1 = maps[i][:]
    m2 = [m[i] for m in maps]
    j1 = judge(m1)
    j2 = judge(m2)
    answer.append(j1)
    answer.append(j2)
print(min(answer))