[SWEA] 5648. 원자 소멸 시뮬레이션

link 🔗 https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWXRFInKex8DFAUo 

SW Expert Academy

 

1️⃣ 두 개 이상의 원자가 충돌 ➡️ 에너지 방출 & 소멸

2️⃣ 상 하 좌 우 중 1 방향으로 1초에 1 움직임

3️⃣ 동시에 이동

 

t = int(input())
directy = [1, -1, 0, 0]
directx = [0, 0, -1, 1]
for tc in range(1, t+1):
    N = int(input())
    arr = [list(map(int, input().split())) for _ in range(N)] # x, y, 방향, 보유 에너지 K
    for idx in range(len(arr)):
        arr[idx][0] *= 2
        arr[idx][1] *= 2
    energy = 0
    for _ in range(4002): # -2000 ~ 2000
        for idx in range(len(arr)):
            arr[idx][0] += directx[arr[idx][2]]
            arr[idx][1] += directy[arr[idx][2]]

        ddel, visit = set(), set()
        for idx in range(len(arr)):
            nx, ny = arr[idx][0], arr[idx][1]
            if (nx, ny) in visit: # ni, nj가 visit에 있으면 -> DEL에 추가
                ddel.add((nx, ny))
            visit.add((nx, ny))

        for idx in range(len(arr)-1, -1, -1):
            if (arr[idx][0], arr[idx][1]) in ddel:
                energy += arr[idx][3]
                arr.pop(idx)

    print(f'#{tc} {energy}')

 

✏️ 메모

• 처음에 바로 답 나와서 와 웬일이지? 했는데 역시나... 테스트 케이스 중 n개만 맞췄습니다...^^ 어쩐지 쉽다 했다..🥲
• 주어진 예시를 보면 -6인 A와 -3인 B는 -4.5에서 부딪히게 되는데, -4.5는 실수이므로 이를 위해 각 좌표에 2씩을 곱해주어 자연수 상태에서 부딪힐 수 있게 해주어야 했다.
• 또 주어진 좌표 값이 y, x가 아니라 x, y라는 점...ㅎㅎ 분명 맞게 했는데 자꾸 fail 떠서 한참 들여다 봤더니 이동시켜주는 과정에서 x와 y를 바꿔 적은 것이 원인이었다.
• 배열에서는 상 하가 각각 y-1, y+1이었지만 이 문제에서 상 하는 y+1, y-1이다.