[코딩테스트]/[SQL]

[프로그래머스 Lv.5] 멸종위기의 대장균 찾기

잰잰' 2025. 4. 11. 16:21

문제 설명

대장균들은 일정 주기로 분화하며, 분화를 시작한 개체를 부모 개체, 분화가 되어 나온 개체를 자식 개체라고 합니다.

다음은 실험실에서 배양한 대장균들의 정보를 담은 ECOLI_DATA 테이블입니다. ECOLI_DATA 테이블의 구조는 다음과 같으며, IDPARENT_IDSIZE_OF_COLONYDIFFERENTIATION_DATEGENOTYPE 은 각각 대장균 개체의 ID, 부모 개체의 ID, 개체의 크기, 분화되어 나온 날짜, 개체의 형질을 나타냅니다.

최초의 대장균 개체의 PARENT_ID 는 NULL 값입니다.

 

문제

각 세대별 자식이 없는 개체의 수(COUNT)와 세대(GENERATION)를 출력하는 SQL문을 작성해주세요. 이때 결과는 세대에 대해 오름차순 정렬해주세요. 단, 모든 세대에는 자식이 없는 개체가 적어도 1개체는 존재합니다.

 

예시

예를 들어 ECOLI_DATA 테이블이 다음과 같다면

각 세대별 대장균의 ID는 다음과 같습니다.

1 세대 : ID 1, ID 2
2 세대 : ID 3, ID 4, ID 5
3 세대 : ID 6, ID 7
4 세대 : ID 8

이 때 각 세대별 자식이 없는 대장균의 ID는 다음과 같습니다.

1 세대 : ID 1
2 세대 : ID 3, ID 5
3 세대 : ID 7
4 세대 : ID 8

따라서 결과를 세대에 대해 오름차순 정렬하면 다음과 같아야 합니다.

 

✏️ 풀이

WITH RECURSIVE GEN AS (
    SELECT ID, PARENT_ID, 1 AS 'GENERATION'
    FROM ECOLI_DATA
    WHERE PARENT_ID IS NULL
    
    UNION ALL
    
    SELECT D.ID, D.PARENT_ID, G.GENERATION + 1 AS 'GENERATION'
    FROM ECOLI_DATA AS D
    JOIN GEN AS G ON D.PARENT_ID = G.ID
)

SELECT COUNT(*) AS `COUNT`, GENERATION
FROM GEN AS G
LEFT JOIN ECOLI_DATA AS D ON G.ID = D.PARENT_ID
WHERE D.PARENT_ID IS NULL
GROUP BY 2
ORDER BY 2;

Level5가 두 문제 밖에 안돼서 마지막 문제까지 다 풀었다

쿼리 작성 자체에는 어려움이 없지만 문제를 푸는 방법을 찾는게 매번 어려운 것 같다

이번에 RECURSIVE를 사용해서 재귀쿼리도 처음 써봤는데 자주 써봐야겠다