블로그를 이전 중이라 완료되기 전까지는 벨로그에 작성할 계획입니다.
이후 모든 글은 https://weekwith.me 에 작성 예정이니 다른 글이 궁금하시다면 해당 링크를 통해 방문해주세요.본 글은 [ 백준 ] 1213번: 팰린드롬 만들기를 풀고 작성한 글입니다.
문제
설명
임한수와 임문빈은 서로 사랑하는 사이이다.
임한수는 세상에서 팰린드롬인 문자열을 너무 좋아하기 때문에, 둘의 백일을 기념해서 임문빈은 팰린드롬을 선물해주려고 한다.
임문빈은 임한수의 영어 이름으로 팰린드롬을 만들려고 하는데, 임한수의 영어 이름의 알파벳 순서를 적절히 바꿔서 팰린드롬을 만들려고 한다.
임문빈을 도와 임한수의 영어 이름을 팰린드롬으로 바꾸는 프로그램을 작성하시오.
입력
첫째 줄에 임한수의 영어 이름이 있다. 알파벳 대문자로만 된 최대 50글자이다.
출력
첫째 줄에 문제의 정답을 출력한다. 만약 불가능할 때는 "I'm Sorry Hansoo"를 출력한다. 정답이 여러 개일 경우에는 사전순으로 앞서는 것을 출력한다.
풀이
접근법
팰린드롬이 만들어지는 문자열인지 유효성 검사를 하기 위해 주어진 입력 문자열의 총 개수가 홀수인 경우와 짝수인 경우를 나눠서 생각해야 한다.
홀수인 경우 하나의 문자만 무조건 홀수개이고 나머지 문자는 짝수개여야 한다.
짝수인 경우 모든 문자가 짝수개여야 한다.
이러한 유효성 검사를 거친 뒤 통과한 문자열을 사전순으로 정렬한 뒤 각각 개수의 절반씩 나열하고 이를 뒤집어서 다시 더해주면 된다.
이때 입력 문자열의 총 개수가 홀수였을 경우 중간 문자를 하나 넣어줘야 한다.
나의 풀이
접근법을 토대로 문제를 풀면 아래와 같다.
from collections import defaultdict
S: str = input()
total_count: int = 0
alphas: dict[str, int] = defaultdict(int)
for alpha in S:
total_count += 1
alphas[alpha] += 1
odd_count: int = 0
even_count: int = 0
for alpha, count in alphas.items():
if (count % 2):
odd_count += 1
else:
even_count += 1
if (total_count % 2) and (odd_count != 1):
print("I'm Sorry Hansoo")
elif not (total_count % 2) and (odd_count != 0):
print("I'm Sorry Hansoo")
else:
alphas = sorted(alphas.items(), key=lambda x: x[0])
front_to_end: str = ""
end_to_front: str = ""
center_alpha: str = ""
for alpha, count in alphas:
if count % 2:
front_to_end = front_to_end + (alpha * (count // 2))
end_to_front = (alpha * (count // 2)) + end_to_front
center_alpha = alpha
else:
front_to_end = front_to_end + (alpha * (count // 2))
end_to_front = (alpha * (count // 2)) + end_to_front
answer: str = front_to_end + center_alpha + end_to_front
print(answer)Big-O
sorted() 내장함수에 의해 시간 복잡도는 O(NlogN)이다.
기타
collections 모듈 내에 있는 defaultdict() 함수를 사용해서 문제를 풀었는데 단순히 배열을 만들어서 문제를 풀어도 됐을 것 같다.
그랬으면 시간 복잡도 또한 O(N)으로 줄일 수 있었을 것이다.
