[SQL] leetcode | 180. Consecutive Numbers 예제 풀이

문제

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

 

코드

나의 답안

# Write your MySQL query statement below
with result as (
select *, lag(num) over (order by id) as prev_num
from logs
),
session_marker as (
select
    *,
    (case 
    when prev_num is null then 1
    when num = prev_num then 0
    else 1
    end) as marker
from result
),
session_group as (
select
    id,
    num,
    prev_num,
    sum(marker) over (order by id) as grp
from session_marker
)
select distinct num as ConsecutiveNums
from session_group
where grp in 
    (select grp
    from session_group
    group by grp
    having count(*) >= 3)

-- 5분

 

타 블로그 답안

SELECT DISTINCT num0 AS ConsecutiveNums
FROM (SELECT id
            , num AS num0
            , LEAD(num, 1) OVER (ORDER BY id) AS num1
            , LEAD(num, 2) OVER (ORDER BY id) AS num2
        FROM Logs) logs
WHERE num0=num1 AND num1=num2

 

문제 풀이

어떤 조건을 만족하는 행이 연속적인지 판별하는 문제이다.

 

위의 방법으로 처음 학습을 해서 그런가

비슷한 문제만 보면 세션 마킹 -> 세션 그룹 생성 -> 세션 그룹 별 행 count 이런 식으로 풀게 된다.

 

문제는 그렇게 풀지 않아도 될 것들도 이렇게 푸니까 시간만 잡아먹을 때가 있음

 

이 문제도 최소 3개 연속만 충족하는 숫자만 구하면 되기 때문에

다음 행, 다다음 행을 동시에 구하고 비교하면 된다.

 

내가 사용한 방법은 결과값에 조건이 더 많이 들어갈 때 사용하는 게 좋다.

(참고 : https://syimmin-data-analysis.tistory.com/12)

(이 예제에서는 세션마다 시간의 최소, 최대값을 구해야 했다)

 

느낀 점

문제를 읽고 어떤 문법을 사용하는 게 적절할지 생각해보고 코드를 짜자!