[SWEA][S/W 문제해결 기본][Python] 1220 Magnetic

[문제]

https://swexpertacademy.com/main/talk/solvingClub/problemView.do?solveclubId=AV6kld8aisgDFASb&contestProbId=AV14hwZqABsCFAYD&probBoxId=AV-4MojKLNADFATz+&type=PROBLEM&problemBoxTitle=%5BD2%7ED3+%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4%5D+%EA%B8%B0%EC%B4%88+%EB%8B%A4%EC%A7%80%EA%B8%B0+Part4&problemBoxCnt=++14+

SW Expert Academy

 

 

 

[코드]

def dfs(arr, n):
    cnt = 0

    for j in range(n):
        visit = False
        for i in range(n):
            if arr[i][j] == '1':  #N극일 경우 
                visit = True	#visit을 True로 전환
            elif arr[i][j] == '2' and visit: #N극이 나오고 S극이 나왔을 때 
                cnt += 1
                visit = False 	#cnt + 1 해주고 visit을 다시 False로 전환해준다. 

    return cnt


#main
for tc in range(1,11):
    n = int(input())  #정사각형 한 변의 길이
    arr = [input().split() for _ in range(n)]

    res = dfs(arr, n)
    print(f'#{tc} {res}')