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

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

1213번: 팰린드롬 만들기

 

문제

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

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

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

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

입력

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

출력

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

 

---

내 풀이

# 31120kb 68ms

알파벳 순서가 적용된 팰린드롬이란게 이 문제의 특징이었다. 

딕셔너리에 알파벳에 따라 숫자를 정리하고,

이 때 홀수 개인 알파벳이 2개 이상이면 팰린드롬이 불가능하기 때문에 -1을 출력하고 종료.

그 외에는 홀 수개가 있는 알파벳 1개를 저장하고 (odd_string)

모든 알파벳의 1/2 개를 알파벳 순서대로 정렬 (ans)

최종 출력하면서 팰린드롬을 완성했다 (ans + odd_string + ans(뒤집어서))

string = input() # 예시: AAABB
pel_dict = {}
cnt = 0
odd_string = ''
ans = ''

for i in string:
  if i in pel_dict:
    pel_dict[i] += 1
  else:
    pel_dict[i] = 1

for j in pel_dict.values(): # 홀수 글자가 2개 이상이면 팰린드롬 불가 cnt
  if j % 2 == 1:
    cnt += 1

if cnt > 1:
  print("I'm Sorry Hansoo")
else:
  pel_list = sorted(pel_dict.items(), key=lambda x: x[0])     # [('A',3), ('B',2)]
  for key, value in pel_list:
    if value % 2 == 1:
      odd_string = key           # 홀수개인 A 하나 빼주고
      ans += key * (value//2)    # 알파벳 순서로 팰린드롬 배치
    else:
      ans += key * (value//2)

  print(ans + odd_string + ans[::-1])   # AB + A + BA

 

메모

그리디 알고리즘