[TIL] MySQL TIL Day 5

🟡 [HackerRank] Weather Observation Station 20

https://www.hackerrank.com/challenges/weather-observation-station-20/problem?isFullScreen=true

Solution

이 문제는 단순히 ‘median 함수를 써야지!’ 라는 생각으로 접근했다간 큰일난다. Oracle에서는 MEDIAN 함수를 제공하지만 mySQL에서는 이 함수를 제공하지 않아 직접 median을 구해야 한다.

이 문제를 해결하기 위해서는 데이터를 lat_n을 기준으로 오름차순 정렬 후, 각 행에 행 번호를 나타내는 새로운 변수를 하나 생성하면 된다. 그 후 Median을 구하는 수식을 이용해 정답을 찾는다.

SET

mySQL에서 SET은 새로운 변수를 할당할 때 쓰는 문법이다. 주어진 데이터의 변수만으로는 문제를 해결할 수 없는 경우, SET을 이용해 필요한 변수를 생성해주면 된다. 이 문제 같은 경우는 행 번호를 생성해야 하므로 다음과 같이 설정한다.

예시

SET @rownum = -1; % 세미콜론 필수
SELECT @rownum := rownum + 1 AS rnum
FROM station

위와 같이 코드를 실행하면 rownum이 0부터 시작하여 $0, 1, 2, 3, \dots$ 과 같이 생성된다. 이러한 방법으로 행 번호를 0부터 차례대로 지정해준다. 단, 이번 문제에서는 lat_n을 먼저 오름차순으로 정렬한 후 행 번호를 부여해야 한다. 풀이는 아래 코드에 있다.

Code

맨 아래 WHERE 절이 바로 median을 찾는 코드인데, 이 때 median에 해당하는 행 번호를 찾기 위해서는 @rownum 을 이용한다. @rownum := @rownum + 1 에서 최종적으로 마지막 행의 번호가 @rownum에 할당되었기 때문에 이를 사용해줘야 한다.

SET @rownum = -1; % 세미콜론 필수
SELECT ROUND(AVG(lat_n), 4) AS median
FROM (SELECT @rownum := @rownum + 1 AS rnum, lat_n
      FROM station
      ORDER BY lat_n) AS stat
WHERE rnum IN (FLOOR(@rownum/2), CEIL(@rownum/2))