PCCP 모의고사 1회차 - 3번 문제
코딩테스트 준비
해당 썸네일은 Wonkook Lee 님이 만드신 Thumbnail-Maker 를 이용하였습니다
⛳️ 문제
멘델은 완두콩을 이용하여 7년간 실험한 결과, 다음과 같은 특별한 법칙을 발견하였습니다.
둥근 완두 순종(RR)을 자가 수분, 즉 같은 유전자끼리 교배할 경우, 다음 세대에 둥근 완두 순종 형질만 나타난다.
주름진 완두 순종(rr)을 자가 수분할 경우, 다음 세대에 주름진 완두 순종 형질만 나타난다. 두 순종을 교배한 잡종(Rr)을 자가 수분할 경우, 다음 세대의 형질은 RR:Rr:rr=1:2:1의 비율로 나타난다. (아래 그림 참조)
멘델의 법칙을 공부한 진송이는, 직접 완두콩의 자가 수분 실험을 진행했습니다. 진송이의 실험에서 완두콩 한 개를 자가 수분한 결과는 다음과 같습니다.
각 완두콩은 자가 수분해서 정확히 4개의 완두콩 후손을 남긴다.
잡종 완두콩(Rr)은 자가 수분해서 첫째는 RR, 둘째와 셋째는 Rr, 넷째는 rr 형질의 후손을 남긴다.
순종 완두콩(RR, rr)은 자가 수분해서 자신과 같은 형질의 후손을 남긴다. 잡종 완두콩(Rr) 1대부터 시작한 가계도로 그려보면 그림 2와 같습니다.
진송이는 이러한 완두콩의 자가 수분 실험 결과를 정리하고 싶어합니다. 하지만, 세대를 거듭할수록, 완두콩의 수가 너무 많아져 모든 가계도를 기록하기 어려워졌습니다. 진송이는 가계도를 전부 기록하는 것 대신, 완두콩의 세대와 해당 세대에서 몇 번째 개체인지를 알면 형질을 바로 계산하는 프로그램을 만들려 합니다.
각 세대에서 맨 왼쪽 개체부터 첫 번째, 두 번째, 세 번째, …개체로 나타냅니다. 예를 들어 그림 2에서 2세대의 네 번째 개체의 형질은 “rr”이며, 3세대의 9번째 개체의 형질은 “RR”입니다.
형질을 알고 싶은 완두콩의 세대를 나타내는 정수 n과, 해당 완두콩이 세대 내에서 몇 번째 개체인지를 나타내는 정수 p가 2차원 정수 배열 queries의 원소로 주어집니다.
queries에 담긴 순서대로 n세대의 p 번째 개체의 형질을 문자열 배열에 담아서 return 하도록 solution 함수를 완성해주세요.
🥄 솔루션
- 완두콩 x의 형질은 부모의 형질을 알아야 한다.
- p를 통해 완두콩 x가 부모의 몇 번째 자식인지를 파악한다.
- 조건 중 무조건 4개의 완두콩 후손을 남기기 때문에 파악을 할 수 있다.
- 결국 x의 시초가 어떤 유전 형질을 가졌는지를 알면 순서에 따른 x의 형질을 결정 지을 수 있다.
- 그럼 재귀적으로 위로 타고 올라가면서 몇 번째 후손인지 알아내야 한다.
코드
def get_ans(stack):
while stack:
c_idx = stack.pop()
if c_idx == 0:
return "RR"
if c_idx == 3:
return "rr"
return "Rr"
def solution(queries):
answer = []
case = len(queries)
for tc in range(case):
n, p = map(lambda x:x-1, queries[tc])
stack = []
while n>0:
p, c = divmod(p, 4)
stack.append(c)
n -= 1
answer.append(get_ans(stack))
return answer
분석 및 후기
- 시간 복잡도 : O(q * n)
- 공간복잡도 : O(q + n)
어렵게 생각했다가 오히려 간단한 문제였다. 이 문제에서 집중해야 할 부분은 깔끔하게 짜는 것이다.
스택을 통해 현재 상태를 관리하고 divmod, 함수형 프로그래밍을 통해 가독성과 리턴되는 값들을 처리하는 것이 읽는 사람도 짜는 사람도 편안해 할 것이다.
