[SWEA] Magnetic

[SWEA] Magnetic

Problem URL : Magnetic

for tc in range(1, 11):
    N = int(input())
    arr = [list(map(int, input().split())) for _ in range(N)]
    ans = 0
    for i in range(N):
        q = []
        for j in range(N):
            if not q:
                if arr[j][i] == 1:  # 비어 있을 때는 N극을 추가
                    q.append(1)
            elif q[-1] + arr[j][i] == 3:  # 다른 극일 때 큐에 추가
                q.append(arr[j][i])
        ans += len(q)//2  # N극, S극 2개씩 뭉텅이 하나에 교착상태 1개

    print('#{} {}'.format(tc, ans))