[Python] SWEA D3 : 1220 - [S/W 문제해결 기본] 5일차 - Magnetic

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV14hwZqABsCFAYD&categoryId=AV14hwZqABsCFAYD&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=2

SW Expert Academy

---

📌 작성한 코드 

for test in range(1,11):
    n = int(input())
    arr = [[x for x in input().split()] for _ in range(n)]
    turn = list(map(list,zip(*arr)))
    temp = []
    ans = 0
    for i in turn:
        temp.append("".join(i).replace('0',''))
    for i in temp:
        ans += i.count('12')
    print(f'#{test} {ans}')

 

 

📌 풀이 

1. 배열의 세로줄만 추출하기 

turn = list(map(list,zip(*arr)))

 

2. 0을 제거하고 1과 2로만 이루어진 문자열로 만들기 

for i in turn:
    temp.append("".join(i).replace('0',''))

 

3. '12' 문자열을 찾기(s극와 n극에 이끌리는 자석들이 서로 부딪혀 교착상태가 되면 12가 붙어있게 됨)

for i in temp:
    ans += i.count('12')