15-oesnuj

[oesnuj]

🔗 문제 링크

백준 | 주식

✔️ 소요된 시간

1시간 30분

✨ 수도 코드

그리디 문제를 많이 풀어본 것 같지 않아 한번 파보기로 했습니다. 문제집에서 제목이 제일 끌리는 문제로 골라봤습니다.

시간초과때문에 한번에 맞추지는 못한 문제였습니다.
그냥 답을 출력하는 것은 문제가 아니었지만 그리디 답게 더 효율적으로 찾는 법을 더 생각해내야했네요

😢문제를 보자마자 생각이 든 풀이(시간초과)

1. 날 별 주가 배열에서 최댓값을 찾는다.
2. 최댓값보다 앞에 위치하면서 작은 모든 값을 배열에서 제거 시킨다.
3. 다시 1번으로 돌아가 다음 최댓값을 찾는다.

위 로직을 배열이 빌때까지 반복

function calc(i){
    let stockFlow = input[i].trim().split(' ').map(e => parseInt(e));
    let profit = 0;

    while(stockFlow.length > 0){
        const max = Math.max(...stockFlow);
        const maxIndex = stockFlow.indexOf(max);

        for(let j = maxIndex; j >= 0; j--) {
            if (max >= stockFlow[j]) {
                profit += max - stockFlow[j];
                stockFlow.splice(j, 1);
            }
        }
    }
    return profit;
}

시간 초과 이유를 생각해보면 새로운 최댓값을 계속 찾아내는 과정과 splice()로 배열에서 제거시키는 로직이 문제였던 것 같습니다.

😃 통과한 풀이

더 효율적인 방법이 뭘까 이것저것 계속 고민했었는데요....
배열에서 제거시킬때 배열을 뒤에서 부터 순회해야 제거된 인덱스가 다음 순회에 영향을 주지 않았습니다.
여기서 나온 생각이었는데 그냥 뒤에서 부터 최댓값을 찾습니다.
순회 중에 현재까지의 가장 비싼 주가가 남아있는데요.
이 가격보다 비싸게 파는 날를 만나기 전까지는 모두 현재까지의 최대 가격으로 팔아주면 최고 익절이 됩니다.

최적화 로직:

• 최대 주가 찾기: 뒤에서부터 현재까지의 주가 중 가장 비싼 날을 찾습니다.
• 이익 계산: 현재 주가와 그 이후의 최고 주가 간의 차이를 이익으로 계산합니다.

const filepath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const input = require('fs').readFileSync(filepath).toString().trim().split('\n'); //모든 입력 \n으로 파싱

let answer ="";
for(let i = 2; i < input.length; i += 2) { //날 별 주가가 적혀있는 인덱스만 추출
    answer += calc(i) + "\n";
}
console.log(answer);

function calc(i){
    let stockFlow = input[i].split(' ').map(Number); //날 별 주가 흐름을 배열로 만듦
    let profit = 0;
    let maxPrice = 0;
    for(let j = stockFlow.length - 1; j >= 0; j--) {
        if (stockFlow[j] > maxPrice) {
            maxPrice = stockFlow[j];
        }
        profit += maxPrice - stockFlow[j];
    }
    return profit;
}

📚 새롭게 알게된 내용

이런 유형에서 뒤에서부터 순회하는 방식을 이용하니까 엄청 간단하게 짜지네요.
그동안 한번도 생각 못해본 아이디어였는데 하나 알아갑니다.

현재까지의 최댓값을 처리하는 느낌에서 모노톤 스택를 이용해서 풀어도 괜찮을 것 같아요