[백준] 20056. 마법사 상어와 파이어볼 / python 파이썬

🚩 시뮬레이션, 구현

* 삼성 SW 역량 테스트 기출 문제

 

thinking

"격자의 행과 열은 1번부터 N번까지 번호가 매겨져 있고, 1번 행은 N번과 연결되어 있고, 1번 열은 N번 열과 연결되어 있다."

난 이게 뭔말인가 했다;; 😇

이거였음!!!!!!

빨간별이 파이어볼

파이어볼 방향

이런 격자판에서 문제에서 하라는데로만 순서대로 진행하면 된다.

예제 1을 예시로 들어보면 이렇다

 

예제 1

(1,1)파이어볼은 속력이 2이므로 2번 방향으로 두칸가고, (1,4)파이어볼은 속력이 1이므로 6번 방향으로 1칸 간다.

(1,3)에서 두개 이상의 파이어볼이 만났으므로, [2-1] 조건대로 하나의 파이어볼로 합쳐지고, [2-2] 조건대로 4개의 파이어볼로 나누어진다. [2-3] 조건대로 각 4개의 파이어볼의 질량은 (5+7) / 5 = 2 이고, 방향은 합쳐지는 파이어볼 방향이 2, 6 으로 모두 짝수이므로 0,2,4,6 이 된다.

 

 

코드

N, M, K = map(int, input().split())
fireballs = []
for _ in range(M):
    _r, _c, _m, _s, _d = list(map(int, input().split()))
    fireballs.append([_r-1, _c-1, _m, _s, _d])

MAP = [[[] for _ in range(N)] for _ in range(N)]

dx = [-1, -1, 0, 1, 1, 1, 0, -1]
dy = [0, 1, 1, 1, 0, -1, -1, -1]

for _ in range(K):
    # 파이어볼 이동
    while fireballs:
        cr, cc, cm, cs, cd = fireballs.pop(0)
        nr = (cr + cs * dx[cd]) % N  # 1번-N번 행 연결되어있기 때문
        nc = (cc + cs * dy[cd]) % N
        MAP[nr][nc].append([cm, cs, cd])

    # 2개 이상인지 체크
    for r in range(N):
        for c in range(N):
            # 2개 이상인 경우 -> 4개의 파이어볼로 쪼개기
            if len(MAP[r][c]) > 1:
                sum_m, sum_s, cnt_odd, cnt_even, cnt = 0, 0, 0, 0, len(MAP[r][c])
                while MAP[r][c]:
                    _m, _s, _d = MAP[r][c].pop(0)
                    sum_m += _m
                    sum_s += _s
                    if _d % 2:
                        cnt_odd += 1
                    else:
                        cnt_even += 1
                if cnt_odd == cnt or cnt_even == cnt:  # 모두 홀수이거나 모두 짝수인 경우
                    nd = [0, 2, 4, 6]
                else:
                    nd = [1, 3, 5, 7]
                if sum_m//5:  # 질량 0이면 소멸
                    for d in nd:
                        fireballs.append([r, c, sum_m//5, sum_s//cnt, d])

            # 1개인 경우
            if len(MAP[r][c]) == 1:
                fireballs.append([r, c] + MAP[r][c].pop())

print(sum([f[2] for f in fireballs]))