[BOJ/Python] 백준 3048번 - 개미

백준 #3048번 개미

3048번: 개미

import sys
input = sys.stdin.readline

n1, n2 = map(int, input().split())   # 첫 번째 그룹 개미 수, 두 번째 그ㄹ룹 개미 수
n1_ants =list(input().strip())
n2_ants = list(input().strip())
t = int(input())

for i in range(len(n1_ants)):   # 첫 번째 그룹임 표시
  n1_ants[i] = (n1_ants[i], 1)
for i in range(len(n2_ants)):   # 두 번째 그룹임 표시
  n2_ants[i] = (n2_ants[i], 2)

pre_line = n1_ants[::-1] + n2_ants
after_line = n1_ants[::-1] + n2_ants
for _ in range(t):
  for i in range(len(pre_line)-1):
    if pre_line[i][1] == 1 and pre_line[i+1][1] == 2:
      after_line[i], after_line[i+1] = after_line[i+1], after_line[i]
  pre_line = after_line[:]

for i in range(len(after_line)):
  print(after_line[i][0], end='')

첫 번째 그룹과 두 번째 그룹을 하나의 리스트로 만든 후, 첫 번째 그룹 개미 앞에 두 번째 그룹 개미가 있을 경우 두 개미의 자리를 교체한다. 첫 번째 그룹인지 두 번째 그룹인지 표시하기 위해 각 요소를 그룹의 번호도 들어있는 튜플로 바꿔주었다. 그룹을 이은 리스트를 두 개 만든 이유는 두 개미가 자리를 교체했을 때 방금 교체된 개미 앞에 다른 그룹이 있어 또 교체되는 상황이 발생할 수 있기 때문에 두 개의 리스트로 분리해주었다.

 

 

---

 

그룹을 구분하기 위해 그룹 번호가 들어있는 튜플로 요소들을 변경해주었는데, 이렇게 하지 않고 그룹 번호만 담고 있는 리스트를 따로 만들어서 관리해주었으면 더 보기 편할 것 같다. group = [1, 1, 1, 2, 2, 2] 이런 식으로!