[SWEA] 1220. Magnetic - Python

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV14hwZqABsCFAYD 

SW Expert Academy

 

< 📝 문제 >
테이블 위에 자성체들이 놓여 있다.

자성체들은 성질에 따라 색이 부여되는데, 푸른 자성체의 경우 N극에 이끌리는 성질을 가지고 있고, 붉은 자성체의 경우

 

S극에 이끌리는 성질이 있다.

아래와 같은 테이블에서 일정 간격을 두고 강한 자기장을 걸었을 때, 시간이 흐른 뒤에 자성체들이 서로 충돌하여 테이

 

블 위에 남아있는 교착 상태의 개수를 구하라.

아래는 자성체들이 놓여 있는 테이블을 위에서 바라본 모습이다.

A로 표시된 붉은 자성체의 경우 S극에 이끌리면서 테이블 아래로 떨어지게 된다.

B로 표시된 푸른 자성체의 경우 N극에 이끌리면서 테이블 아래로 떨어지게 된다.

나머지 자성체들은 서로 충돌하며, 교착 상태에 빠져 움직이지 않게 된다.

D로 표시된 자성체들에서 알 수 있듯 한쪽 방향으로 움직이는 자성체의 개수가 많더라도 반대 방향으로 움직이는 자성

 

체가 하나라도 있으면 교착 상태에 빠져 움직이지 않는다.

D로 표시된 자성체들과 같이 셋 이상의 자성체들이 서로 충돌하여 붙어 있을 경우에도 하나의 교착 상태로 본다.

C와 D는 좌우로 인접하여 있으나 각각 다른 교착 상태로 판단하여 2개의 교착 상태로 본다.

E의 경우와 같이 한 줄에 두 개 이상의 교착 상태가 발생할 수도 있다.

F의 경우 각각 다른 교착상태로 판단하여 2개의 교착상태로 본다.

 

위의 예시의 경우 테이블 위에 남아있는 교착상태는 7개이므로 7을 반환한다.

< ❓ 생각 >
빨간색(1)은 아래로 파란색(2)은 위로 끌려간다

빨간색이 위, 파란색이 아래에 있으면 서로 반대방향으로 가려고 해서 충돌하여 교착 상태에 빠진다

따라서, 열을 기준으로 볼 때 빨간색 다음에 파란색이 나오면 교착의 개수 +1

 

먼저 열 방향으로 순회시켜서 그 열에 교착이 나올 때마다 +1해주기

열을 이동하면서 모든 열 확인

 

< 💻 코드 >

# 테스트 케이스 개수
T = 10
for tc in range(1,11):
    # 정사각형 테이블의 한 변의 길이
    N = int(input())
    # 테스트 케이스
    arr = [list(map(int,input().split())) for _ in range(N)]
    
    # 빨간색(1) -> S극
    # 파란색(2) -> N극
    # 열을 기준으로 볼 때
    # 빨간색(1) 나오고 파란색(2) 나오면 교착
    cnt = 0
    for c in range(N):
        isRed = False
        for r in range(N):
            if arr[r][c] == 1:
                isRed = True
            if arr[r][c] == 2 and isRed:
                cnt += 1
                isRed = False
    print(f'#{tc} {cnt}')

< ❗ 느낀 점 >

생각이 바로 떠오르지 않아 어려운 문제였다. 단순하게 빨간색 다음 파란색이 오기만 하면 교착인데 너무 꼬아서 생각해서 오래 걸린 문제.ㅜ