[SQL] leetcode | 601. Human Traffic of Stadium 예제 풀이 | 어떤 조건을 만족하는 행이 연속적으로 n개

문제

https://leetcode.com/problems/human-traffic-of-stadium/description/

 

 

WITH result AS (
    SELECT 
        id, 
        visit_date, 
        people,
        LAG(people) OVER (ORDER BY id) AS prev_people
    FROM stadium
),
session_marker AS (
    SELECT
        id,
        visit_date,
        people,
        (CASE
            WHEN prev_people IS NULL THEN 1
            WHEN prev_people < 100 THEN 1
            WHEN people < 100 THEN 1
            ELSE 0
        END) AS marking
    FROM result
),
session_grouping AS (
    SELECT
        id,
        visit_date,
        people,
        SUM(marking) OVER (ORDER BY id) AS session_group
    FROM session_marker
)
SELECT 
    id, 
    visit_date, 
    people
FROM session_grouping
WHERE session_group IN (
    SELECT session_group
    FROM session_grouping
    GROUP BY session_group
    HAVING COUNT(DISTINCT id) >= 3
)
ORDER BY session_group, id;

-- 40분 .. ㅎ

^ 장장 40분에 걸쳐 완성한 내 구구절절 코드 .. ㅎ

 

WITH session_group AS (
    SELECT 
        *,
        ROW_NUMBER() OVER (ORDER BY id) AS rownum,
        id - ROW_NUMBER() OVER (ORDER BY id) AS grp
    FROM Stadium
    WHERE people >= 100
)
SELECT 
    sg.id,
    sg.visit_date,
    sg.people
FROM session_group sg
WHERE sg.grp IN (
    SELECT grp
    FROM session_group
    GROUP BY grp
    HAVING COUNT(*) >= 3)

^ 타 블로그에서 퍼온 천재적인 코드 ..!

 

이 정도면 수학/논리적 사고의 싸움인 듯 ..

 

문제 풀이

session_group CTE : 

session_group CTE를 확인해보면
1차로 people >= 100 으로 걸러낸 테이블에서

id - row number를 구하면

100이 넘는 연속적인 행은 같은 grp 숫자를 가지게 된다.

 

그 다음에 grp의 행 갯수가 3개 이상인 걸 출력하면 되겠다

 

해당 풀이 방법은 순차적인 id가 존재하는 테이블에 활용할 수 있겠다

 

느낀 점

지난번 세선 재정의하기 문제와 비슷해보여서 같은 원리로 코드를 짰고 성공하긴 했지만 시간이 너무 오래 걸렸다.

결국 이 문제를 풀 수 있는 주요 원리는 달랐다.

 

해당 풀이 방법을 이번에 기억해서 다른 문제에 써먹을 수 있게 해야겠다 !

그리고 다양한 문제를 풀어보는 게 중요하다는 걸 실감했다