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

문제

20056번 마법사 상어와 파이어볼

📌 아이디어

구현, 시뮬레이션 문제이기에 요구사항에 맞는 조건을 순차적으로 해결 및 작성하는 방식으로 문제를 해결했습니다.

1. 2차원 배열에 파이어볼의 질량, 속도, 방향 정보를 리스트 형태로 저장
2. 파이어볼이 이동할 위치를 저장하는 배열 생성 및 위치 이동 후 저장
3. 이동이 완료된 파이어볼을 탐색하며 한 위치에 파이어볼이 2개인 경우 조건 처리
4. 최종적으로 파이어볼이 있는 경우 질량을 누적

📌 코드

N, M, K = map(int, input().split())
arr = [[[] for _ in range(N)] for _ in range(N)]
for _ in range(M):
    r, c, m, s, d = map(int, input().split())
    arr[r-1][c-1].append([m, s, d])

# 0 ~ 7까지의 방향
dir = [(-1, 0), (-1, 1), (0, 1), (1, 1), (1, 0), (1, -1), (0, -1), (-1, -1)]

for _ in range(K):
    # 이동 할 위치를 저장할 배열
    board = [[[] for _ in range(N)] for _ in range(N)]
    for i in range(N):
        for j in range(N):
            if arr[i][j] != []:
                # 위치 이동
                for m, s, d in arr[i][j]:
                    r = (i + dir[d][0] * s) % N
                    c = (j + dir[d][1] * s) % N
                    board[r][c].append([m, s, d])

    # 이동한 파이어볼들을 체크
    for i in range(N):
        for j in range(N):
            # 한 위치에 파이어볼이 2개 이상인 경우
            if len(board[i][j]) >= 2:
                mass = 0;speed = 0;check1 = 0;check2 = 0
                # 질량과 속도 구하기 + 홀짝 체크
                for m, s, d in board[i][j]:
                    mass += m
                    speed += s
                    if (d % 2)==0:check1 += 1
                    if (d % 2): check2 += 1

                mass //= 5
                # 질량이 0인 경우 위치를 초기화하고 continue
                if mass == 0:
                    board[i][j] = []
                    continue
                speed //= len(board[i][j])
                # 홀수나 짝수인 경우
                if check1 == len(board[i][j]) or check2 == len(board[i][j]):
                    board[i][j] = []
                    for d in [0, 2, 4, 6]:
                        board[i][j].append([mass, speed, d])
                # 그 외의 경우
                else:
                    board[i][j] = []
                    for d in [1, 3, 5, 7]:
                        board[i][j].append([mass, speed, d])
    # 2차원 배열를 통해
    arr = [lst[:] for lst in board]

ans = 0
# 파이어볼이 있는 경우 질량 구하기
for i in range(N):
    for j in range(N):
        if arr[i][j] != []:
            for m, s, d in arr[i][j]:
                ans += m

print(ans)코드를 입력하세요

피드백 및 개선점은 댓글을 통해 알려주세요😊