백준 알고리즘 3048번 개미(python)

같이 스터디하시는 분의 코드를 참고했다. 클래스를 사용해서 풀 생각은 하지 못했는데 코드를 보자마자 감명받았다.

_ = map(int, input().split())
g1 = [{'group': 1, 'name': ant} for ant in input()][::-1]
g2 = [{'group': 2, 'name': ant} for ant in input()]
ants = g1 + g2
T = int(input())

for _ in range(T):
    i = 0
    while i < len(ants) - 1:
        if ants[i]['group'] < ants[i + 1]['group']:
            ants[i], ants[i + 1] = ants[i + 1], ants[i]
            i += 1
        i += 1
        
print("".join([ant['name'] for ant in ants]))