[LeetCode] 601. Human Traffic of Stadium

📢 본 포스팅에 활용되는 기본 문제 및 자료 출처는

       리트코드 Problems, https://leetcode.com/problemset/all/임을 밝힙니다.

---

■ 문제

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

Human Traffic of Stadium - LeetCode

Table : Stadium

Write an SQL query to display the records with three or more rows with consecutive id's, and the number of people is greater than or equal to 100 for each.

Return the result table ordered by visit_date in ascending order.

The query result format is in the following example.

---

■ 풀이

  문제 요구사항  

아이디가 3개 이상 연속이면서 각 방문자 수가 100 이상인 행을 출력하시오.

즉, 위 아래 행의 방문자 수가 100 이상인 행을 출력하시오.이전행, 다음행을 출력할 수 있는 LEAD(), LAG() 함수를 이용할 수 있다.단, 원래 테이블의 모든 행이 기준행이 될 수 있으므로 기준행 앞 2행, 뒤2행을 모두 따져주어야 한다.

 

문제 조건을 충족하는 경우의 수는 총 세가지로 나뉜다.

---

(id컬럼 오름차순 기준)

기준행-2 people값	기준행-1 people값	기준행 people값	기준행+1 people값	기준행+2 people값
A	B	C	D	E

  ✔ case1 | C와 A, B가 각각 100명 이상인 경우

  ✔ case2 | C와 B, D가 각각 100명 이상인 경우

  ✔ case3 | C와 D, E가 각각 100명 이상인 경우

---

case는 아래와 같이 WHERE절에 조건으로 넣어줄 수 있다. 

WITH T1 AS (SELECT id, visit_date, people
    ,lead(people, 1) OVER (order by id) AS da1_people
    ,lead(people, 2) OVER (order by id) AS da2_people
    ,lag(people, 1) OVER (order by id) AS db1_people
    ,lag(people, 2) OVER (order by id) AS db2_people
    FROM stadium)

SELECT id, visit_date, people
 FROM T1
 WHERE T1.people >= 100 AND ((T1.da1_people >= 100 AND T1.da2_people >= 100)
                        OR (T1.db1_people >= 100 AND T1.da1_people >= 100)
                        OR (T1.db1_people >= 100 AND T1.db2_people >= 100))

---

✏ 개인회고

처음에 LEAD함수만 사용해서 마지막 id 2행이 people값이 100 이상 임에도 출력되지 않는 것에서 막혔다.

아래 2행을 따로 출력할 방법만 고민하다가 기준행 앞2개 뒤2개까지 컬럼으로 만들어주고 보니 조건에 해당하는 행을 누락없이 출력할 수 있었다. 

WHERE 조건절도 기존에 AND, OR만 사용하다가 경우의 수로 나누는 조건을 적용해볼 수 있는 문제였다.

 문제에서 한번 사용한 문법은 잘 기억해둘 것!