https://www.acmicpc.net/problem/11501
난이도
실버2
문제
홍준이는 요즘 주식에 빠져있다. 그는 미래를 내다보는 눈이 뛰어나, 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다. 매일 그는 아래 세 가지 중 한 행동을 한다.
주식 하나를 산다.
원하는 만큼 가지고 있는 주식을 판다.
아무것도 안한다.
홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.
예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.
입력
입력의 첫 줄에는 테스트케이스 수를 나타내는 자연수 T가 주어진다. 각 테스트케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고, 둘째 줄에는 날 별 주가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다. 날 별 주가는 10,000이하다.
출력
각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다.
내 코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringBuilder sb = new StringBuilder();
StringTokenizer st = new StringTokenizer(br.readLine());
int T = Integer.parseInt(st.nextToken());
for (int tc = 0; tc < T; tc++) {
int day = Integer.parseInt(br.readLine());
int[] num = new int[day];
st = new StringTokenizer(br.readLine());
for (int i = 0; i < day; i++) {
num[i] = Integer.parseInt(st.nextToken());
}
long get = 0;
int max = num[day - 1];
for (int i = day - 1; i >= 0; i--) {
if (num[i] < max) {
get += max - num[i];
} else {
max = num[i];
}
}
sb.append(get).append("\n");
}
System.out.println(sb);
}
}
최대한 많은 점수를 얻어야 하기 때문에 뒤에서부터 확인하며 앞으로 넘어오도록 구현했다.
뒤에서 앞으로 오면서 값이 점점 작아지는 내리막 길 같은 점수라면 낮은 주가에서 사서 제일 큰 주가에서 파는게 이득이다.
max 값을 지정해두고 max보다 작다면 max - 현재 주가 가 이득으로 돌아오기 때문에 get에 이익을 더하면 된다.
만약 현재 max 값보다 더 큰 값을 만났다면 이득이 되지 않기 때문에 max값을 그 값으로 변경해주고 그냥 넘어가면 된다.
4% 틀렸던 이유는, 답은 부호있는 64bit 정수형으로 표현 가능 이 부분 때문이었다. get을 int로 선언했어서 범위를 넘어갔고, long으로 변경해주니 통과했다.
