[SWEA] 2382. 미생물 격리 (Python)

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

SW Expert Academy

 

 

t = int(input())
for tc in range(1, t+1):
    N, M, K = map(int, input().split()) # 셀의 개수, 격리 시간, 군집의 개수
    misaeng = [list(map(int, input().split())) for _ in range(K)] # 세로, 가로, 미생물 수, 이동방향
    direct = [(-1, 0), (1, 0), (0, -1), (0, 1)]  # 상 하 좌 우
    modi_dir = []
    ret = 0
    for _ in range(M):
        for m in misaeng:
            # 미생물 이동, 경계처리
            m[0] = m[0]+direct[m[3]-1][0] # 세로 이동
            m[1] = m[1]+direct[m[3]-1][1] # 가로 이동
            if m[0] == 0 or m[0] == N-1 or m[1] == 0 or m[1] == N-1: # 약품이 발라져 있는 곳에 다다르면
                m[2] //= 2
                if m[3] % 2 == 1:
                    m[3] += 1 # 방향이 홀수면 1 더해주고
                else:
                    m[3] -= 1 # 방향이 짝수면 1을 빼주면 반대 방향이 된다
        # 내림차순 정렬 후 같은 좌표에 있다면 병합
        misaeng.sort(key=lambda x:(x[0], x[1], x[2]), reverse=True) # 세로, 가로, 미생물 순서로 내림차순 정렬
        idx = 1
        while idx < len(misaeng):
            if misaeng[idx-1][0] == misaeng[idx][0] and misaeng[idx-1][1] == misaeng[idx][1]: # y, x 좌표가 같으면
                misaeng[idx-1][2] += misaeng[idx][2] # 앞의 미생물 수에(내림차순 정렬했으니까 더 큼) 현재 미생물 수 더해주고
                misaeng.pop(idx) # 현재 군집 삭제
            else: # 좌표 같지 않으면 idx 1 더해주기
                idx += 1
    for s in misaeng: # 남은 미생물 수 모두 더해주기
        ret += s[2]
 
    print(f'#{tc} {ret}')

 

💥 주의

• 미생물들 한 번씩 이동시킨 다음 정렬 → 같은 값 병합했어야 했는데, 이동할 때마다 정렬해서 같은 값 병합하고 있었다...ㅎㅎ 어쩐지 뭐가 이상하더라니
• 으아아ㅏㅏ 알고리즘 잘하고 싶다ㅏㅏㅏ