백준(BOJ) Implement 🐜 기적의 매매법 🐜(20546) - Silver 5

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

20546번: 🐜 기적의 매매법 🐜

문제

"오늘도 호재만 있게 해주세요. 버핏-"

2년차 개미 준현이는 오늘도 버핏신에게 기도를 올린다. 장기 투자를 지향하는 준현이는 한 번 산 주식은 절대 팔지 않는다. 2099년이 되어도 주식을 팔지 않을 것이다. 주식 매수 후 오로지 기도만 하기 때문에 이를 BNP 전략이라고 한다. BNP는 Buy and Pray의 약자이다. 준현이는 주식을 살 수 있다면 무조건 최대한 많이 산다. 준현이는 욕심쟁이이기 때문에, 주식을 살 수 있다면 가능한 만큼 즉시 매수한다. 다음은 준현이가 현금 100원으로 A기업의 주식을 사는 경우이다.

 1일2일3일4일5일6일7일현금주가매수 가능 주식 수매수여부남은 현금보유 주식 수

100	20	20	6	0	0	0
40	33	7	2	1	12	50
2	0	2	3	0	0	0
O	X	O	O	X	X	X
20	20	6	0	0	0	0
2	2	4	7	7	7	7

"주식은 타이밍이지!"

반면, 성민이는 오늘도 주식 거래 프로그램을 실행한다. 모니터 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"을 출력한다. 모든 결과 따옴표를 제외하고 출력한다.

 

문제 해결

시물레이션 구현 문제이다. 시물레이션 문제는 처음 로직을 잘 구성한 후 그 로직에만 맞게 코드를 구현하면 쉽게 해결은 가능하다. 코드 설계에 충분한 시간을 두고 진행해야 구현 단계에서 쉽게 문제를 해결할 수 있을 것 같다... 일단 이 문제에서 준현이는 단순히 첫날부터 구매가 가능하면 주식을 구매하고 마지막날 주식의 가격에 맞게 전량 매도하면 된다. 문제는 성민이의 매매 패턴인데 성민이는 다음과 같은 특징이 있다.

1. 3일 연속 하락장이면 다음날은 무조건 상승장이다. 따라서 무조건 전량 매수

2. 3일 연속 상승장이면 다음날은 무조건 하락장이다. 따라서 무조건 전량 매도

3. 가격이 같은 날은 상승장도 하락장도 아니다.

 

위 패턴에 맞게 구현해주면 된다.

• 3일 연속 하락장인지 확인 -> 구매해야함

bool buypossible(vector<ll>&days, ll index){
    ll testcase = 2; // 현재 날짜 이전부터 최근 3일간 가격이 하락한 경우
    ll idx = index-1;
    while(testcase--){
        if(days[idx] < days[idx-1])
            idx--;
        else
            return false;
    }
    return true;
}

• 3일 연속 상승장인지 확인 -> 판매해야함

bool sellpossible(vector<ll>&days, ll index){ // 현재 날짜 이전부터 최근 3일간 가격이 상승한 경우
    ll testcase = 2;
    ll idx = index-1;
    while(testcase--){
        if(days[idx] > days[idx-1])
            idx--;
        else
            return false;
    }
    return true;
}

현재 날짜부터 3일만 확인하면 된다.  이렇게 조건에 맞게 전체 코드를 구성해주면 된다.

 

전체 코드

#include <iostream>
#include <algorithm>
#include <vector>
#define SIZE 14

using namespace std;
typedef long long ll;

ll BNP(ll money, vector<ll>&days){
    ll stock = 0;
    for(ll i=1; i<=SIZE; i++){
        if(money == 0)
            break;
        else{
            stock = stock +(money/days[i]);
            money = money % days[i];
        }
    }
    return money + stock*days[SIZE];
}  

bool buypossible(vector<ll>&days, ll index){
    ll testcase = 2; // 현재 날짜 이전부터 최근 3일간 가격이 하락한 경우
    ll idx = index-1;
    while(testcase--){
        if(days[idx] < days[idx-1])
            idx--;
        else
            return false;
    }
    return true;
}

bool sellpossible(vector<ll>&days, ll index){ // 현재 날짜 이전부터 최근 3일간 가격이 상승한 경우
    ll testcase = 2;
    ll idx = index-1;
    while(testcase--){
        if(days[idx] > days[idx-1])
            idx--;
        else
            return false;
    }
    return true;
}


ll TIMING(ll money, vector<ll>&days){
    ll i = 0;
    ll stock = 0;
    for(i=4; i<= SIZE; i++){
        if(buypossible(days, i) == true && money != 0){ // 돈이 있고 3일 연속 가격이 하락세라면 전량 구매
            stock = stock +(money/days[i]);
            money = money % days[i];
        }
        else if(sellpossible(days, i) == true && stock != 0){ // 주식이 있고 3일 연속 상승세라면 전량 매도
            money += stock * days[i];
            stock = 0;
        }
    }
    return money + (stock * days[SIZE]);
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    // freopen("input.txt", "r" ,stdin);

    ll money = 0;
    cin >> money;

    vector<ll>days(SIZE+1, 0);
    for(ll i=1; i<=SIZE; i++)
        cin >> days[i];

    ll bnp = BNP(money, days);
    ll timing = TIMING(money, days);

    if(bnp < timing)
        cout << "TIMING\n";
    else if(bnp > timing)
        cout << "BNP\n";
    else
        cout << "SAMESAME\n";
    return 0;
}