[백준 20546 / Silver5] 🐜 기적의 매매법 🐜 - Java(자바)/Swift(스위프트)

❓문제 설명

작성자 : 김민규
문제 링크 : https://www.acmicpc.net/problem/20546

• "오늘도 호재만 있게 해주세요. 버핏-"
  2년차 개미 준현이는 오늘도 버핏신에게 기도를 올린다. 장기 투자를 지향하는 준현이는 한 번 산 주식은 절대 팔지 않는다. 2099년이 되어도 주식을 팔지 않을 것이다. 주식 매수 후 오로지 기도만 하기 때문에 이를 BNP 전략이라고 한다. BNP는 Buy and Pray의 약자이다. 준현이는 주식을 살 수 있다면 무조건 최대한 많이 산다. 준현이는 욕심쟁이이기 때문에, 주식을 살 수 있다면 가능한 만큼 즉시 매수한다. 다음은 준현이가 현금 100원으로 A기업의 주식을 사는 경우이다.
  "주식은 타이밍이지!"
  반면, 성민이는 오늘도 주식 거래 프로그램을 실행한다. 모니터 8개에서 뿜어져 나오는 화려한 주식 차트가 성민이를 감싼다. 성민이는 주식이 타이밍 싸움이라 생각한다. 전형적인 단기 투자자로 생각하면 오산이다. 성민이만의 전략이 있기 때문이다. 이른바 33 매매법으로, 그 방법은 다음의 세 가지 룰로 이루어져있다.
  • 모든 거래는 전량 매수와 전량 매도로 이루어진다. 현재 가지고 있는 현금이 100원이고 주가가 11원이라면 99원어치의 주식을 매수하는 것이다. 단, 현금이 100원 있고 주가가 101원이라면 주식을 살 수 없다. 성민이는 빚을 내서 주식을 하지는 않는다.
  • 3일 연속 가격이 전일 대비 상승하는 주식은 다음날 무조건 가격이 하락한다고 가정한다. 따라서 현재 소유한 주식의 가격이 3일째 상승한다면, 전량 매도한다. 전일과 오늘의 주가가 동일하다면 가격이 상승한 것이 아니다.
  • 3일 연속 가격이 전일 대비 하락하는 주식은 다음날 무조건 가격이 상승한다고 가정한다. 따라서 이러한 경향이 나타나면 즉시 주식을 전량 매수한다. 전일과 오늘의 주가가 동일하다면 가격이 하락한 것이 아니다.
    준현이와 성민이는 각자의 매매법 중 어떤 방법이 더 수익률이 높은지 겨뤄보기로 했다. 오로지 MachineDuck이라는 기업의 주식만 거래가 가능하며, 내기 기간은 2021년 1월 1일부터 2021년 1월 14일까지이다. 준현이와 성민이에게 주어진 현금은 동일하다. 세기의 대결이기 때문에 이 기간에는 매일 주식 거래가 가능하다. 2021년 1월 14일에 더 많은 자산을 보유한 사람이 승리한다. 1월 14일의 자산은 (현금 + 1월 14일의 주가 × 주식 수)로 계산한다.
    우리는 2021년 1월 1일부터 2021년 1월 14일까지의 주식 가격을 미리 알고 있다. 준현이와 성민이 중 누가 더 높은 수익률을 낼지 맞혀보자!

❗입출력

[입력]
첫 번째 줄에 준현이와 성민이에게 주어진 현금이 주어진다.
두 번째 줄에 2021년 1월 1일부터 2021년 1월 14일까지의 MachineDuck 주가가 공백을 두고 차례대로 주어진다. 모든 입력은 1000 이하의 양의 정수이다.
[출력]
1월 14일 기준 준현이의 자산이 더 크다면 "BNP"를, 성민이의 자산이 더 크다면 "TIMING"을 출력한다.
둘의 자산이 같다면 "SAMESAME"을 출력한다. 모든 결과 따옴표를 제외하고 출력한다.
[예시]

🎁 문제 풀이

• 준현이의 BNP는 주어진 금액을 이용해서 최대한 많이 구매합니다.
  • 보유한 현금으로 주식의 가격을 나눴을 때 0보다 크면 해당 주식을 모두 매수합니다.
• 성민이의 TIMING은 3일 연속으로 하락했을 경우에만 구매합니다.
  • 3일 연속으로 하락하고 다음날의 주식을 구매하므로 i+3번째 가격으로 매수합니다.
  • 3일 연속으로 상승하면 다음날에 해당 주식을 i+3번째 가격으로 매도합니다. (주식 보유시에만)
  • 14일 후의 준현이와 성민이의 현금과 마지막 날 주식 가격에 보유 주식 수를 곱해서 비교합니다.

🖥️ 코드

Java

import java.util.*;
import java.io.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int[] arr = new int[14];

        StringTokenizer st = new StringTokenizer(br.readLine()," ");
        for (int i = 0; i < 14; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        int JHmoney = N;
        int JHstock = 0;

        for (int i = 0; i < arr.length; i++) {
            if (JHmoney / arr[i] > 0) {
                int cnt = JHmoney / arr[i];
                JHstock += cnt;
                JHmoney -= cnt * arr[i];
            }
        }

        int SMmoney = N;
        int SMstock = 0;

        for (int i = 0; i < arr.length - 3; i++) {
            if ((arr[i] > arr[i + 1]) && (arr[i + 1] < arr[i + 2])) {
                if (SMstock == 0) {
                    continue;
                }
                SMmoney += arr[i + 3] * SMstock;
                SMstock = 0;
            }
            else if ((arr[i] < arr[i + 1]) && (arr[i + 1] > arr[i + 2])) {
                if (SMmoney / arr[i + 3] > 0) {
                    int cnt = SMmoney / arr[i + 3];
                    SMstock += cnt;
                    SMmoney -= cnt * arr[i + 3];
                }
            }
        }

        int JHreturn = JHmoney + arr[arr.length - 1] * JHstock;
        int Smreturn = SMmoney + arr[arr.length - 1] * SMstock;

        if (JHreturn > Smreturn) {
            System.out.println("BNP");
        } else if (JHreturn < Smreturn) {
            System.out.println("TIMING");
        } else {
            System.out.println("SAMESAME");
        }
    }
}

swift

import Foundation

let N = Int(readLine()!)!
let arr = readLine()!.split(separator: " ").map { Int($0)! }

var JHmoney = N
var JHstock = 0

for i in 0..<arr.count {
    if JHmoney / arr[i] > 0 {
        let cnt = JHmoney / arr[i]
        JHstock += cnt
        JHmoney -= cnt * arr[i]
    }
}

var SMmoney = N
var SMstock = 0

for i in 0..<arr.count - 3 {
    if arr[i] < arr[i + 1] && arr[i + 1] < arr[i + 2] {
        if SMstock == 0 {
            continue
        }
        SMmoney += SMstock * arr[i + 3]
        SMstock = 0
    }
    else if arr[i] > arr[i + 1] && arr[i + 1] > arr[i + 2] {
        if SMmoney / arr[i + 3] > 0 {
            let cnt = SMmoney / arr[i + 3]
            SMstock += cnt
            SMmoney -= cnt * arr[i + 3]
        }
    }
}

let JHreturn = JHmoney + arr.last! * JHstock
let SMreturn = SMmoney + arr.last! * SMstock

if JHreturn > SMreturn {
    print("BNP")
} else if JHreturn < SMreturn {
    print("TIMING")
} else {
    print("SAMESAME")
}