[백준] 1213번 - 팰린드롬 만들기

www.acmicpc.net

https://www.acmicpc.net/problem/1213

문제

💡

임한수와 임문빈은 서로 사랑하는 사이이다.

임한수는 세상에서 팰린드롬인 문자열을 너무 좋아하기 때문에, 둘의 백일을 기념해서 임문빈은 팰린드롬을 선물해주려고 한다.

임문빈은 임한수의 영어 이름으로 팰린드롬을 만들려고 하는데, 임한수의 영어 이름의 알파벳 순서를 적절히 바꿔서 팰린드롬을 만들려고 한다.

임문빈을 도와 임한수의 영어 이름을 팰린드롬으로 바꾸는 프로그램을 작성하시오.

시간 제한: 2초

메모리 제한: 128MB

입력

💡

첫째 줄에 임한수의 영어 이름이 있다. 알파벳 대문자로만 된 최대 50글자이다.

출력

💡

첫째 줄에 문제의 정답을 출력한다. 만약 불가능할 때는 "I'm Sorry Hansoo"를 출력한다. 정답이 여러 개일 경우에는 사전순으로 앞서는 것을 출력한다.

예제 입력	예제 출력
AABB	ABBA
AAABB	ABABA
ABACABA	AABCBAA
ABCD	`I'm Sorry Hansoo`

풀이

팰린드롬은 거꾸로 읽어도 제대로 읽는 것과 같은 문장이나 낱말, 숫자, 문자열 등을 말한다.

중복되는 문자가 많거나 문자열이 길어지면 경우의 수가 여러개이지만, 사전순으로 앞서는것 하나만 출력하라 했기 때문에 탐욕알고리즘을 사용하면 쉽게 풀 수 있을 것이다.

우선 사전순으로 출력하기 위해 입력 배열을 내림차순으로 정렬한다. 이후 연속되는 똑같은 문자를 따로 저장하고, 홀짝 여부를 확인해 정답 배열에 삽입한다. 홀수이면 중간에 하나를 저장한뒤 좌, 우에 저장하고, 짝수면 바로 좌, 우에 저장한다.

이렇게 저장한 정답배열이 팰린드롬의 규칙에 위배되는지 확인한다.

코드


#include <iostream>
#include <algorithm>
#include <deque>

using namespace std;

deque<char> rearrange(string& input) {
	deque<char> tmp;
	deque<char> ans;
	
	for (int i = 0; i < input.size(); ++i) {
		tmp.push_back(input[i]);

		if (tmp.back() == input[i + 1]) {
			continue;
		}
		else {
			if (tmp.size() % 2 != 0) {
				ans.emplace(ans.begin() + (ans.size() / 2), tmp.front());
				tmp.pop_front();
			}
			for (int j = 0; j < tmp.size() && tmp.size() != 0; j += 2) {
				ans.push_front(tmp[j]);
				ans.push_back(tmp[j + 1]);
			}
			tmp.clear();
		}
	}
	return ans;
}

bool verify(const deque<char>& q) { // 12345 123456
	deque<char> a(q.size()/2); //
	deque<char> b(q.size()/2);
	copy(q.begin(), q.begin() + q.size() / 2, a.begin()); //12
	copy(q.end() - q.size() / 2, q.end(), b.begin()); //345
	sort(b.begin(), b.end());
	return equal(a.begin(), a.end(), b.begin(), b.end());
}

int main() {
	string input;
	cin >> input;
	sort(input.begin(), input.end(), greater<char>());
	deque<char> ans = rearrange(input);

	if (verify(ans)) {
		for (auto i : ans) {
			cout << i;
		}
	}
	else {
		cout << "I'm Sorry Hansoo";
	}

	return 0;
}