[C++/백준] 11501번-주식

JG's Development Blog·2022년 8월 25일
0

코딩 문제풀이

목록 보기
10/32

링크


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

문제


홍준이는 요즘 주식에 빠져있다. 그는 미래를 내다보는 눈이 뛰어나, 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다. 매일 그는 아래 세 가지 중 한 행동을 한다.

  1. 주식 하나를 산다.
  2. 원하는 만큼 가지고 있는 주식을 판다.
  3. 아무것도 안한다.
    홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.

예를 들어 날 수가 3일이고 날 별로 주가가 10, 7, 6일 때, 주가가 계속 감소하므로 최대 이익은 0이 된다. 그러나 만약 날 별로 주가가 3, 5, 9일 때는 처음 두 날에 주식을 하나씩 사고, 마지막날 다 팔아 버리면 이익이 10이 된다.

입력


입력의 첫 줄에는 테스트케이스 수를 나타내는 자연수 T가 주어진다. 각 테스트케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고, 둘째 줄에는 날 별 주가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다. 날 별 주가는 10,000이하다.

출력


각 테스트케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 답은 부호있는 64bit 정수형으로 표현 가능하다.

풀이


순방향, 역방향 두가지의 방법으로 풀었다.

먼저 순방향 방법은,
vector와 algorithm의 max_element를 사용하여 저장된 값의 최댓값을 구해준다음 순방향으로 차례로 검사하면서 현재 인덱스의 값과 최댓값이 같다면 다음 인덱스부터 끝까지의 값 중에서 최댓값을 다시 구하고 그 외의 경우는 result값에 (최댓값 - 현재 검사중인 값)을 더해주었다.

결론적으로 위 방법도 맞았지만 8%에서 자꾸 틀려서 역방향의 방법으로도 해보았다.

역방향 방법은 더 간단하다.
맨 뒤의 원소값을 최댓값으로 잡고 뒤에서부터 검사하여 최댓값보다 작으면 result값에 더해주고, 아닌 경우 max를 현재 검사중인 값으로 갱신해주면 된다.

이 방법 또한 8%에서 틀리는 바람에 찾아본 결과 문제에 제시된 '답은 부호있는 64bit 정수형으로 표현 가능하다.'는 말은 결국 result의 타입이 int가 아닌 long long으로 사용되었어야 했던 것이다.

8%에서 틀리는 분들은 result의 타입이 long long으로 설정되었는지 확인해보면 좋을 것이다.

코드


  1. 순방향 방법
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

int main() {
	int T;
	cin >> T;

	vector <int> v;
	int N, num;
	int max;
	int result;
	for (int i = 0; i < T; i++) {
		cin >> N;
		result = 0;
		v.clear();
		for (int j = 0; j < N; j++) {
			cin >> num;
			v.push_back(num);
		}

		max = *max_element(v.begin(), v.end());
		for (int j = 0; j < N; j++) {
			if (v[j] == max && v[j] != v.back()) {
				max = *max_element(v.begin() + j + 1, v.end());
			}
			else {
				result += max - v[j];
			}
		}
		cout << result << endl;
	}

	return 0;
}
  1. 역방향 방법
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

int main() {
	int T;
	cin >> T;

	vector <int> v;
	int N, num;
	int max;
	long long result;
	for (int i = 0; i < T; i++) {
		cin >> N;
		result = 0;
		v.clear();
		for (int j = 0; j < N; j++) {
			cin >> num;
			v.push_back(num);
		}

		max = v[N - 1];
		for (int j = N - 1; j >= 0; j--) {
			if (v[j] <= max) {
				result += max - v[j];
			}
			else {
				max = v[j];
			}
		}
		cout << result << endl;
	}

	return 0;
}

profile
왕왕왕초보

0개의 댓글