Programmers / [PCCP 모의고사 1회] 3번 / 유전법칙 / python

 

 

문제: https://school.programmers.co.kr/learn/courses/15008/lessons/121685

 

풀이

def solution(queries):
    def find(n, p):
        if n == 1:
            return "Rr"

        parent = find(n-1, (p-1)//4+1)
        idx = (p-1) % 4

        if parent == "RR":
            return "RR"
        elif parent == "rr":
            return "rr"
        else:
            return ["RR", "Rr", "Rr", "rr"][idx]

    return [find(n, p) for n, p in queries]

 

부모의 위치를 찾아서 (p-1)//4 + 1 로 계산

부모의 형질에 따라 현재 개체의 형질이 결정되므로 부모 형질에 따른 유전 규칙이 적용되도록 한다.