코딩테스트연습(파이썬)
숫자 짝꿍 / 프로그래머스, 파이썬 코딩테스트 연습 연습문제
LearnerToRunner
2022. 12. 19. 21:59
문제
두 정수 X, Y의 임의의 자리에서 공통으로 나타나는 정수 k(0 ≤ k ≤ 9)들을 이용하여 만들 수 있는 가장 큰 정수를 두 수의 짝꿍이라 합니다(단, 공통으로 나타나는 정수 중 서로 짝지을 수 있는 숫자만 사용합니다). X, Y의 짝꿍이 존재하지 않으면, 짝꿍은 -1입니다. X, Y의 짝꿍이 0으로만 구성되어 있다면, 짝꿍은 0입니다.
예를 들어, X = 3403이고 Y = 13203이라면, X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 3, 0, 3으로 만들 수 있는 가장 큰 정수인 330입니다. 다른 예시로 X = 5525이고 Y = 1255이면 X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 2, 5, 5로 만들 수 있는 가장 큰 정수인 552입니다(X에는 5가 3개, Y에는 5가 2개 나타나므로 남는 5 한 개는 짝 지을 수 없습니다.)두 정수 X, Y가 주어졌을 때, X, Y의 짝꿍을 return하는 solution 함수를 완성해주세요.
제출답안
def solution(X, Y):
def zip_num_count(str):
"""
스트링 내 글자와 빈도수를 딕셔너리로 리턴하는 함수
"""
dict_str = {}
for n in str:
try:
dict_str[n] += 1
except:
dict_str[n] = 1
return dict_str
# X, Y를 {글자:카운트}로 변환
dict_x, dict_y = zip_num_count(X), zip_num_count(Y)
# X, Y에 공통으로 나오는 글자
commons = dict_x.keys() & dict_y.keys()
# Commons가 비지 않았을 때
if commons:
if commons == {'0'}: # Commons가 0만 있을 때
answer = '0'
else: # Commons가 0외에 다른 숫자도 있을 때
answer = ''
for c in commons:
cnt = dict_x[c] if dict_x[c] <= dict_y[c] else dict_y[c] #최소 카운트
answer += c*cnt
answer = ''.join(sorted(answer, reverse=True))
# Commons가 비었을 때
else:
answer = '-1'
return answer
제출 후 개선답안
a. 딕셔너리 형태 대신 count 활용하여 글자 붙이는 과정 간소화
b. 결과물을 sort하는 대신 큰 수부터 체크하여 answer에 붙여가기
def solution(X, Y):
answer = ''
for i in range(9, -1, -1):
i_str = str(i)
answer += i_str * min(X.count(i_str), Y.count(i_str))
if answer =='': answer = '-1'
elif answer.replace('0', '') == '': answer = '0'
return answer
카운트의 복잡도가 크다고 생각하여
딕셔너리 자료형태로 접근하였었음.
하지만 생각보다 복잡도가 크지 않은듯함.
문제 바로가기
728x90