[Algorithm/Java] 백준 20546번 - 🐜기적의 매매법🐜

---

https://www.acmicpc.net/problem/20546

---

🔍 문제 풀이

문제 도식화

💻 전체 코드

클래스 생성하는 것에 더 익숙해져야 할 것 같다.

클래스 사용

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

public class Main {
    static int[] stockChart;

    static class Investor {
        int cash;
        int stock;

        // 생성자
        public Investor(int cash) {
            this.cash = cash;
            this.stock = 0; // 보유 주식 0으로 초기화
        }

        // 매수
        public void buy(int todayPrice) {
            if (todayPrice <= cash) {
                int count = cash / todayPrice;
                stock += count;
                cash -= count * todayPrice;
            }
        }

        // 매도
        public void sell(int todayPrice) {
            cash += stock * todayPrice;
            stock = 0;
        }

        // 자산 계산
        public int asset(int currentPrice) {
            return cash + stock * currentPrice;
        }
    }


    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int cash = Integer.parseInt(br.readLine());

        stockChart = new int[14];

        StringTokenizer st = new StringTokenizer(br.readLine());

        for(int i=0; i<14; i++){
            stockChart[i] = Integer.parseInt(st.nextToken());
        }
        int bnp = BNP(cash);
        int timing = TIMING(cash);

        if (bnp < timing) System.out.println("TIMING");
        else if (timing < bnp) System.out.println("BNP");
        else System.out.println("SAMESAME");
    }

    static int BNP(int cash){
        Investor investor = new Investor(cash); // 객체 생성
        //주식 절대 안팔아, 살 수 있으면 최대한 많이 매수
        for(int i = 0; i < 14; i++){
            if(stockChart[i] <= cash){
                investor.buy(stockChart[i]); // 전량매수
                investor.asset(stockChart[i]); // 현재 자산
            }
        }

        return investor.asset(stockChart[13]);
    }

    // 성민이는 타이밍
    // 1. 3일연속 주식 상승하면 전량 매도
    // 2. 3일연속 주식 하락하면 전량 매수

    static int TIMING(int cash){
        Investor investor = new Investor(cash); // 객체 생성
        for(int i = 3; i < 14; i++){
            // 3일 연속 주식 상승
            if(stockChart[i-1] > stockChart[i-2] && stockChart[i-2]> stockChart[i-3]){
                // 전량 매도
                investor.sell(stockChart[i]);
            }
            // 3일 연속 주식 하락
            else if(stockChart[i-1] < stockChart[i-2] && stockChart[i-2] < stockChart[i-3]){
                // 전량 매수
                investor.buy(stockChart[i]);
            }
        }
        return investor.asset(stockChart[13]);
    }
}

클래스 미사용

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

public class Main {
    static int[] stockChart;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int cash = Integer.parseInt(br.readLine());
        stockChart = new int[14];

        // 입력
        StringTokenizer st = new StringTokenizer(br.readLine());
        for(int i=0; i<14; i++){
            stockChart[i] = Integer.parseInt(st.nextToken());
        }

        // 호출
        int bnp = BNP(cash);
        int timing = TIMING(cash);

        // 출력
        if (bnp < timing) System.out.println("TIMING");
        else if (timing < bnp) System.out.println("BNP");
        else System.out.println("SAMESAME");

    }

    static int BNP(int cash){
        // 무조건 돈되는대로 전량매수
        int stock = 0;
        for(int i=0; i<14; i++){
            if(cash >= stockChart[i]){
                int buy = cash/stockChart[i]; // 오늘 살 수 있는 주식 수
                stock += buy; // 누적 주식
                cash -= buy * stockChart[i]; // 오늘 산 만큼만 차감
            }
        }
        return cash + stockChart[13] * stock;
    }

    static int TIMING(int cash){
        int stock = 0;
        for(int i=3; i<14; i++){
            // 주식 3일 연속 상승
            if(stockChart[i-1] > stockChart[i-2] && stockChart[i-2] > stockChart[i-3]){
                // 전량 매도
                cash += stock * stockChart[i];
                stock = 0;
            }
            // 주식 3일 연속 하락
            else if(stockChart[i-1] < stockChart[i-2] && stockChart[i-2] < stockChart[i-3]){
                // 전량 매수
                int buy = cash/stockChart[i]; // 오늘 살 수 있는 주식 수
                stock += buy; // 누적 주식
                cash -= buy * stockChart[i]; // 오늘 산 만큼만 차감
            }
        }
        return cash + stockChart[13] * stock;
    }
}