SWEA: 1220 Magnetic

문제

SW Expert Academy

 

문제 풀이

 테이블 위에 N, S극의 성질을 뛰는 자성체가 있을 때 자기장을 가할 경우 N, S극이 만나 더 이상 움직이지 못하는 교착상태가 되는 경우를 반환하는 문제이다. 처음에 문제를 접하였을 때는 N극 S극의 영향을 모두 주어 변화시켜야 한다고 생각하였는데 다시 생각해보면, N극 S극 중 하나만 영향을 받는다고 생각하고 이동을 시키면 된다.

 

예를 들어 S극에 이끌리는 N극 자성체를 이동하고자 하면, 아무것도 없는 경우 이동하고 S극인 자석과 만나는 경우 `교착상태`가 된다.

 

코드

EMPTY, N, S = 0, 1, 2

for test_case in range(1, 11):
    answer = 0
    length = int(input())
    array = [list(map(int, input().split())) for _ in range(length)]

    for i in range(length - 1):
        for j in range(length):
            if array[i][j] == N:
                # 아래칸이 S극이면, 교착 상태
                if array[i + 1][j] == S:
                    answer += 1
                elif array[i + 1][j] == EMPTY:
                    array[i + 1][j] = array[i][j]

    print('#' + str(test_case), answer)