[Leet code] 1393. Capital Gain/Loss

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

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

---

✔문제

https://leetcode.com/problems/capital-gainloss/description/

LeetCode - The World's Leading Online Programming Learning Platform

Table : Stocks

Write a solution to report the Capital gain/loss for each stock.

The Capital gain/loss of a stock is the total gain or loss after buying and selling the stock one or many times.

Return the result table in any order.

The result format is in the following example.

---

✔ 풀이

  문제 요구사항  

operation 컬럼은 '판매(Sell)' 또는 '구매(Buy)'를 값으로 갖는다
각 행은 stock_name을 갖는 각 주식이 Operation_day에 해당 price로 거래되었음을 나타낸다.

각 주식의 자본 이득/손실을 출력해야 한다.

 

① 주식별로 쉽게 집계하기 위해 우선 FROM절 서브쿼리에 주식별 일자순 매수-매도순으로 정렬한 T1 테이블을 만들어주었고

② T1 테이블에서 stock_name을 기준으로 그룹핑해서 전체 매도 price와 전체 매수 price를 CASE문으로 구분한 후 각각의 집계 차이를 capital_gain_loss로 출력했다.

SELECT stock_name, 
        SUM((CASE WHEN operation = 'Sell' THEN price ELSE 0 END)) 
        - SUM((CASE WHEN operation = 'Buy' THEN price ELSE 0 END)) AS capital_gain_loss
 FROM (SELECT *
        FROM Stocks
        ORDER BY 1, 3, 2) T1
 GROUP BY 1

---

✏ 문제풀이 회고

주식별로 전체 매수 합과 매수 합차를 구하는 방식이 아니라 주식별로 거래일자 순으로 매수-매도 값을 순차적으로 계산할 수 있는 방법(함수)이 있을 것 같다. 새로운 2차 풀이도 생각해봐야겠다.

 

-  23.10.05 추가

2차 풀이) CASE문

 손익이라는 개념을 개념대로  -, + 로 생각해보면, CASE문과 집계를 굳이 두번 할 필요가 없었다. 😲 한번의 CASE문에서 매수/매도에 따라 +-로 price 값을 나눠준 후 집계해주면 더 간단하게 풀 수 있다.

SELECT stock_name, 
			SUM(CASE WHEN operation = 'Buy' THEN -price
					ELSE price END
				) AS capital_gain_loss
 FROM Stocks
 GROUP BY 1