[SELECT] 특정 세대의 대장균 찾기

https://school.programmers.co.kr/learn/courses/30/lessons/301650

프로그래머스

---

문제 설명
대장균들은 일정 주기로 분화하며, 분화를 시작한 개체를 부모 개체, 분화가 되어 나온 개체를 자식 개체라고 합니다.
다음은 실험실에서 배양한 대장균들의 정보를 담은 ECOLI_DATA 테이블입니다.

ECOLI_DATA 테이블의 구조는 다음과 같으며, ID, PARENT_ID, SIZE_OF_COLONY, DIFFERENTIATION_DATE, GENOTYPE 은 각각 대장균 개체의 ID, 부모 개체의 ID, 개체의 크기, 분화되어 나온 날짜, 개체의 형질을 나타냅니다.

Column name	Type	Nullable
ID	INTEGER	FALSE
PARENT_ID	INTEGER	TRUE
SIZE_OF_COLONY	INTEGER	FALSE
DIFFERENTIATION_DATE	DATE	FALSE
GENOTYPE	INTEGER	FALSE

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

---

문제
3세대의 대장균의 ID(ID) 를 출력하는 SQL 문을 작성해주세요.

이때 결과는 대장균의 ID 에 대해 오름차순 정렬해주세요.

---

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

ID	PARENT_ID	SIZE_OF_COLONY	DIFFERENTIATION_DATE	GENOTYPE
1	NULL	10	2019/01/01	5
2	NULL	2	2019/01/01	3
3	1	100	2020/01/01	4
4	2	16	2020/01/01	4
5	2	17	2020/01/01	6
6	4	101	2021/01/01	22
7	3	101	2022/01/01	23
8	6	1	2022/01/01	27

PARENT ID 가 NULL 인 ID 1, ID 2가 1 세대이며 ID 1에서 분화된 ID 3, ID 2에서 분화된 ID 4, ID 5 가 2 세대입니다.

ID 4 에서 분화된 ID 6, ID 3에서 분화된 ID 7 이 3 세대이며 ID 6에서 분화된 ID 8은 4 세대입니다.

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

ID

6

7

---

세대별로 나눈 뒤, 특정 세대에 속하는 대장균의 ID를 구하는 문제입니다.

 

부모의 ID로부터 자신이 몇 세대인지 구해야 하므로, 재귀를 사용해 풀어야 하는 문제입니다.

저는 GENERATION이라는 재귀 테이블을 사용해 ID, PARENT_ID, 그리고 GEN을 사용하여 1세대씩 분류하는 방법으로 문제를 풀었습니다.

 

정답 코드를 작성해보겠습니다.

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

  SELECT ID
    FROM GENERATION
   WHERE GEN = 3
ORDER BY ID ASC;

 

처음에는 ECOLI_DATA 테이블에서 ID, PARENT_ID, 1로 테이블을 설정합니다.

이 행으로부터 테이블을 점차 늘려나가는 구조입니다.

 

이후에는 ECOLI_DATA 테이블과 GENERATION 테이블을 이너 조인하여, E 테이블의 부모 ID가 G 테이블의 ID와 동일하다면 세대를 1씩 증가시켜 테이블을 구성합니다.

 

이제 GENERATION 테이블에서 3세대 대장균의 ID를 오름차순으로 출력하면 문제를 풀 수 있습니다.