SQL 예제(Consecutive Numbers)(Lead 함수)

LeetCode 180. Consecutive Numbers

https://leetcode.com/problems/consecutive-numbers/description/

WITH CON AS(
    SELECT NUM,
        LEAD(NUM, 1) OVER() NEXT_ORDER,
        LEAD(NUM, 2) OVER() NEXT_ORDER2
    FROM LOGS
)
SELECT DISTINCT NUM CONSECUTIVENUMS
FROM CON
WHERE (NUM=NEXT_ORDER) AND (NUM=NEXT_ORDER2)

 

Consecutive Number란 연속된 숫자라는 의미다.

예를들어 위와같은 테이블이 있다고했을때

num 컬럼에서 3번 연속으로 동일하게 입력된 숫자를 찾는 것이 문제의 요구사항이다.

어떻게 구할 수 있을까?

 

연속적인 수인지를 판별하기위해서 lead함수를 활용할 수 있다.

lead함수란 특정 컬럼의 특정 행에서 그 다음 행의 값을 확인할 수 있는 함수다.

만약 아래와같이 num이 1, 2, 3인 테이블이 있다고했을때 lead 함수를 적용하면

2, 3, null이 출력된다.

num	next_num
1	2
2	3
3	null

 

lead함수를 예제에 적용해보자

lead함수를 사용하여 num 컬럼 옆에 두번째 열부터 시작하는 컬럼과 세번째 열부터 시작하는 컬럼을 만들어준다.

num과 num_next, num_next2에 모두 같은 값이 들어와있으면 3번 연속 같은 값이 출력됐다고 볼 수 있다.

이를 where절에 쿼리로 작성하면 아래와 같다.

WHERE (NUM=NUM_NEXT) AND (NUM_NEXT=NUM_NEXT2)

 

위의 where절을 반영한 최종 쿼리는 아래와 같이 작성해볼 수 있다.

WITH CON AS(
    SELECT NUM,
        LEAD(NUM, 1) OVER() NUM_NEXT,
        LEAD(NUM, 2) OVER() NUM_NEXT2
    FROM LOGS
)
SELECT DISTINCT NUM CONSECUTIVENUMS
FROM CON
WHERE (NUM=NUM_NEXT) AND (NUM_NEXT=NUM_NEXT2);