백준 14719 빗물 (C++)

14719번: 빗물

간단한 구현 문제이다. 현재 위치를 기준으로 왼쪽과 오른쪽에서 각각 가장 큰 수를 구해 현재 위치의 값과의 차이를 구해준다. 이를 반복하여 결과를 구해주었다.

#include <iostream>
#include <vector>

using namespace std;

int H, W;
vector<int> v;

void solution() {
	int res = 0;

	for (int i = 1; i < W - 1; i++) {
		int left = 0, right = 0;

		for (int j = 0; j < i; j++) {
			left = max(left, v[j]);
		}

		for (int j = W - 1; j > i; j--) {
			right = max(right, v[j]);
		}

		res += max(0, (min(left, right) - v[i]));
	}

	cout << res;
}

int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);

	cin >> H >> W;

	int block;
	for (int i = 0; i < W; i++) {
		cin >> block;
		v.push_back(block);
	}

	solution();
	
	return 0;
}