[백준] 13417번 카드 문자열 파이썬

문제

N장의 카드가 일렬로 놓여있다. 각 카드에는 알파벳이 하나씩 적혀있다. 태욱이는 가장 왼쪽에 있는 카드부터 차례대로 한 장씩 가져올 수 있다. 가장 처음에 가져온 카드는 자신의 앞에 놓는다. 그다음부터는 가져온 카드를 자신의 앞에 놓인 카드들의 가장 왼쪽, 또는 가장 오른쪽에 놓는다. 태욱이는 모든 카드를 다 가져온 후에 자신의 앞에 놓인 카드를 순서대로 이어 붙여 카드 문자열을 만들려고 한다.

예를 들어 3장의 카드가 [M, K, U] 순으로 놓여있다고 하자. 태욱이는 먼저 가장 왼쪽에 있는 “M”이 적힌 카드를 가져와서 자신의 앞에 놓는다. 다음으로 남은 카드 중 가장 왼쪽에 있는 “K”가 적힌 카드를 가져와서 가장 왼쪽에 두고, 이어서 “U”가 적힌 카드를 가져와서 다시 가장 왼쪽에 두면 “UKM”이라는 문자열을 만들 수 있다. 만약 “K”가 적힌 카드를 가져와서 가장 왼쪽에 두고, 이어서 “U”가 적힌 카드를 가져와서 가장 오른쪽에 두면 “KMU”라는 문자열을 만들 수 있다. 이때, 태욱이가 만들 수 있는 문자열 중 사전 순으로 가장 빠른 문자열은 “KMU”이다.

N장의 카드에 적혀있는 알파벳의 처음 순서가 주어질 때, 태욱이가 만들 수 있는 카드 문자열 중 사전 순으로 가장 빠른 문자열을 출력하는 프로그램을 작성하시오.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫째 줄에 테스트 케이스의 개수를 나타내는 자연수 T가 주어진다. 각각의 테스트 케이스의 첫째 줄에 처음에 놓여있는 카드의 개수 N(1 ≤ N ≤ 1,000)이 주어진다. 두 번째 줄에는 N장의 카드에 적힌 알파벳의 초기 순서가 주어진다. 가장 왼쪽에 있는 카드에 적혀있는 알파벳부터 순서대로 N개가 공백으로 구분되어 주어진다. 모든 카드에는 한 개씩의 알파벳이 적혀있으며, 모두 대문자이다.

출력

출력은 표준 출력을 사용한다. 입력받은 데이터에 대해, 한 줄에 1개씩 태욱이가 만들 수 있는 문자열 중에서 사전 순으로 가장 빠른 문자열을 출력한다.

예제 입력 1

3
3
M K U
5
A S D F G
7
B A C A B A C

예제 출력 1

KMU
ASDFG
AAABCBC

문제풀이

• 문자열을 담을 리스트를 선언하고, 전체 횟수 t를 입력받는다.

answer = []
t = int(input())

• t만큼 for loop 돌면서 n과 카드 리스트를 입력받는다.
• answer에 담을 문자열을 입력받은 카드 리스트에서 첫 번재 인덱스의 문자열로 담는다.
  • 맨 처음의 문자열을 가져와야 한다고 했으니 설정해준다.

answer = []
t = int(input())
for _ in range(t):
    n = int(input())
    card_list = input().split()
    temp_str = card_list[0]

• 0번째 인덱스 값을 사용했으니 인덱스 1부터 문자열로 for loop돈다.
  • 현재 값이 temp_str의 맨 왼쪽값보다 크면 가장 오른쪽에, 작거나 같으면 가장 왼쪽으로 두고 temp_str을 재할당 한다.
• 그런 후 answer리스트에 담는다.
• join으로 출력한다.

answer = []
t = int(input())
for _ in range(t):
    n = int(input())
    card_list = input().split()
    temp_str = card_list[0]
    for i in card_list[1:]:
        if ord(i) <= ord(temp_str[0]):
            temp_str = f'{i}{temp_str}'
        else:
            temp_str = f'{temp_str}{i}'
    answer.append(temp_str)
print('\n'.join(answer))

제출답안

import sys

input = sys.stdin.readline


def solution():
    answer = []
    t = int(input())
    for _ in range(t):
        n = int(input())
        card_list = input().split()
        temp_str = card_list[0]
        for i in card_list[1:]:
            if ord(i) <= ord(temp_str[0]):
                temp_str = f'{i}{temp_str}'
            else:
                temp_str = f'{temp_str}{i}'
        answer.append(temp_str)
    print('\n'.join(answer))


solution()