백준 문제/문자열,누적합,구현

[백 준/구현] 1213: 팰린드롬 만들기

goldpig 2023. 12. 26. 22:23
728x90

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

 

1213번: 팰린드롬 만들기

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

www.acmicpc.net

풀이

# 문자의 개수가 홀수이면서 2개 이상이면 팰린드롬을 만들 수 없다.

import sys
input = sys.stdin.readline
from collections import Counter

s = list(input().strip()) # ABAB -> Counter({'A': 2, 'B': 2, '\n': 1}) 개행문자 제거위해 rstrip()
#사전 순으로 출력하기 위해 정렬하기
s.sort()
# 홀수 문자가 몇개인지 확인하기 위해 Counter 함수 사용
check = Counter(s)
# print(check) AAABB -> Counter({'A': 3, 'B': 2})
count = 0
center = ''
# 홀수 문자가 있는지 계산하는 로직
for i in check:
	# print(i, end=' ') A B, print(check[i], end=' ') 3 2
	if check[i] % 2 != 0:
		count += 1
		center += i
		s.remove(i)
	if count >= 2:
		break
#만약 홀수의 개수가 2개 이상이라면 팰린드롬 False
if count >= 2:
	print("I'm Sorry Hansoo")

else:
	res = ""
	# 반복문을 통해 팰린드롬을 반을 나누었을 때 왼쪽 부분을 더해준다.
	for i in range(0, len(s), 2):
		res += s[i]
	print(res + center + res[::-1])

처음에 해결방안을 찾느라 고민을 많이 했었다. 팰린드롬인지 아닌지 판별하는 것은 쉬운 로직이었으나 팰린드롬으로 만드는 로직은 약간 어려웠던거 같다. 로직은 다음과 같다.

  • 사전 순으로 정렬 -> s.sort() 
  • Counter 함수를 통해 홀수 문자가 몇개인지 확인. check = Counter(s)
  • 문자의 개수가 홀수 이면서 2개 이상일 경우에 팰린드롬으로 바꾸지 못함.
  • 홀수의 문자는 center에 넣고 word에서는 제거해줌. center += i, s.remove(i)
  • 홀수의 개수가 1 이하일 경우에는 팰린드롬으로 바꿀 수 있음.
  • 반복문을 통해 팰린드롬을 반으로 나눴을 때에 왼쪽 부분을 res에 더해준다.
  • res + center + res[::-1]을 출력하여 팰린드롬을 완성시킨다.
728x90