문제
N장의 카드가 일렬로 놓여있다. 각 카드에는 알파벳이 하나씩 적혀있다. 태욱이는 가장 왼쪽에 있는 카드부터 차례대로 한 장씩 가져올 수 있다. 가장 처음에 가져온 카드는 자신의 앞에 놓는다. 그다음부터는 가져온 카드를 자신의 앞에 놓인 카드들의 가장 왼쪽, 또는 가장 오른쪽에 놓는다. 태욱이는 모든 카드를 다 가져온 후에 자신의 앞에 놓인 카드를 순서대로 이어 붙여 카드 문자열을 만들려고 한다.
예를 들어 3장의 카드가 [M, K, U] 순으로 놓여있다고 하자. 태욱이는 먼저 가장 왼쪽에 있는 “M”이 적힌 카드를 가져와서 자신의 앞에 놓는다. 다음으로 남은 카드 중 가장 왼쪽에 있는 “K”가 적힌 카드를 가져와서 가장 왼쪽에 두고, 이어서 “U”가 적힌 카드를 가져와서 다시 가장 왼쪽에 두면 “UKM”이라는 문자열을 만들 수 있다. 만약 “K”가 적힌 카드를 가져와서 가장 왼쪽에 두고, 이어서 “U”가 적힌 카드를 가져와서 가장 오른쪽에 두면 “KMU”라는 문자열을 만들 수 있다. 이때, 태욱이가 만들 수 있는 문자열 중 사전 순으로 가장 빠른 문자열은 “KMU”이다.
N장의 카드에 적혀있는 알파벳의 처음 순서가 주어질 때, 태욱이가 만들 수 있는 카드 문자열 중 사전 순으로 가장 빠른 문자열을 출력하는 프로그램을 작성하시오.
입력
입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫째 줄에 테스트 케이스의 개수를 나타내는 자연수 T가 주어진다. 각각의 테스트 케이스의 첫째 줄에 처음에 놓여있는 카드의 개수 N(1 ≤ N ≤ 1,000)이 주어진다. 두 번째 줄에는 N장의 카드에 적힌 알파벳의 초기 순서가 주어진다. 가장 왼쪽에 있는 카드에 적혀있는 알파벳부터 순서대로 N개가 공백으로 구분되어 주어진다. 모든 카드에는 한 개씩의 알파벳이 적혀있으며, 모두 대문자이다.
출력
출력은 표준 출력을 사용한다. 입력받은 데이터에 대해, 한 줄에 1개씩 태욱이가 만들 수 있는 문자열 중에서 사전 순으로 가장 빠른 문자열을 출력한다.
문제 링크
https://www.acmicpc.net/problem/13417
풀이
간단한 그리디 문제다.
정렬로 풀면 되나? 싶지만 문자를 들어온 순서대로, 앞과 뒤에만 배치할 수 있다는 제약이 존재하기에 정렬로는 해결할 수 없다.
대신 매 순간마다 최선의 선택을 하면 된다.
아래의 예시를 보자.
B A C A B A C가 들어왔다고 해보자. ▼
우선 가장 먼저 들어온 B를 문자열의 맨 앞으로 설정을 해준다.
그리고 문자열의 가장 앞을 기준으로 뒤의 문자열들을 사전순으로 비교를 해준다. ▼
이후에는 사진과 같이 문자열의 가장 앞과 비교하며 배치를 해주면 된다. ▼
이렇게 마지막 문자까지 배치가 끝나면 아래와 같이 문자열이 완성된다. ▼
C++ 코드
#include<iostream>
#include<vector>
#include<string>
using namespace std;
int T, N;
string answer;
vector<int> card;
void input() {
char input;
cin >> N;
card = vector<int>(N);
for (int i = 0; i < N; i++) {
cin >> input;
card[i] = int(input);
}
}
void solve() {
answer += char(card[0]);
for (int i = 1; i < N; i++){
if (card[i] <= int(answer[0])) {
answer = char(card[i]) + answer;
} else {
answer = answer + char(card[i]);
}
}
}
void output() {
cout << answer << endl;
answer = "";
}
int main()
{
cin >> T;
while (T--) {
input();
solve();
output();
}
}
'Algorithm > PS' 카테고리의 다른 글
백준 31926번 밤양갱 - C++ (0) | 2024.11.13 |
---|---|
백준 2847번 게임을 만든 동준이 - C++ (0) | 2024.11.12 |
백준 14916번 거스름돈 - C++ (0) | 2024.11.10 |
백준 27961번 고양이는 많을수록 좋다 - C++ (0) | 2024.11.09 |
백준 7569번 토마토 - C++ (0) | 2024.11.08 |