[SW Expert Academy 풀이/ JAVA] 1859.백만장자 프로젝트

이번에 새로 알고리즘 문제를 푸는 사이트를 찾았다.

여기도 생각보다 많은 문제가 있어서 많이 사용할 것 같다.

 

이번 문제는 여기에 들어가서 볼 수 있다.

SW Expert Academy

 

이제 문제풀이로 들어가서 문제를 알려주겠다.

 

문제

25년 간의 수행 끝에 원재는 미래를 보는 능력을 갖게 되었다. 이 능력으로 원재는 사재기를 하려고 한다.

다만 당국의 감시가 심해 한 번에 많은 양을 사재기할 수 없다.

다음과 같은 조건 하에서 사재기를 하여 최대한의 이득을 얻도록 도와주자.

    1. 원재는 연속된 N일 동안의 물건의 매매가를 예측하여 알고 있다.
    2. 당국의 감시망에 걸리지 않기 위해 하루에 최대 1만큼 구입할 수 있다.
    3. 판매는 얼마든지 할 수 있다.

예를 들어 3일 동안의 매매가가 1, 2, 3이라면 처음 두 날에 원료를 구매하여 마지막 날에 팔면 3의 이익을 얻을 수 있다.

 

 

문제만 봤을 때는 잘 모를 거 같다.

예제를 보자.

 

예제

입력

3
3
10 7 6
3
3 5 9
5
1 1 3 1 2

출력

 

#1 0
#2 10
#3 5

음 조건이랑 맞춰보면 대충 최댓값을 잡아 그전까지 구매하고 최댓값 때 팔면 될 것 같다.

 

코드로 옮겨보자.

 

import java.util.Scanner;
class Solution {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        long money = 0;
        long sum = 0;

        int cnt = sc.nextInt();
        sc.nextLine();
        for(int i = 0; i < cnt; i++){
            int count = sc.nextInt();
            sc.nextLine();
            String[] data = sc.nextLine().split(" ");
            int max = max(data,0);
            for(int j = 0; j < count; j++){
                if(max == Integer.parseInt(data[j])){
                    if(j == count - 1 && sum == 0)
                        break;
                    else {
                        data[j] = "0";
                        money += max * sum;
                        max = max(data, j);
                        sum = 0;
                    }
                }else {
                    money -= Integer.parseInt(data[j]);
                    sum++;
                }
            }
            System.out.println("#" + (i+1) + " " + money);
            money = 0;
        }


    }

    public static int max(String[] str,int n){
        int m = 0;
        for (int i = n; i < str.length; i++) {
            if (m < Integer.parseInt(str[i]))
                m = Integer.parseInt(str[i]);
        }
        return m;
    }
}

이번에는 프로그래머스랑 다르게 직접 입력을 받는 거라 Scanner를 사용하였다.

 

먼저 현제 위치에서 최댓값을 가져오는 함수 max를 만들었다.

n번째에서 시작해서 배열 안에 최댓값을 가져오는 함수이다.

내가 값을 String배열로 스캐너의 readline을 이용해 가져왔기에 String []을 매개변수로 받았다.

그다음 지금 값이 max값이면 money에 최댓값과 현재 물건의 개수인 sum을 더해주고, sum을 0으로 초기화해준다.

그다음 다시 그 위치에서부터 max값을 잡아준다.

 

max값이 아니라면 money에서 지금 값을 빼주고, sum을 +해준다.

루프 문이 끝나면 출력해주고, money값을 초기화해 준다.

 

생각보다 쉬운 문제여서 나름 빨리 풀었던 것 같은 문제였다.