[BOJ 16788 // C++] 日本沈没 (Japan Sinks)

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 16788번 문제인 日本沈没 (Japan Sinks)이다.
문제는 아래 링크를 확인하자.

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

16788번: 日本沈没 (Japan Sinks)

해수면이 점점 차오르면 높이가 낮은 땅부터 물에 잠기게 되므로, 땅의 높이순으로 각 땅에 접근해 문제를 해결하자.

 

만약 새로 잠기는 땅의 좌우에 땅이 있다면 섬의 개수가 늘어나고, 한쪽이 바다와 인접해있다면 섬의 개수는 변하지 않으며, 양쪽 다 바다라면 섬이 하나 적어진다는 점을 이용해 단계마다 섬의 개수를 계산해나가자.

 

한편, 같은 높이의 땅은 동시에 잠긴다는 점에 유의하여 시뮬레이션을 진행하자.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int N;
int arr[100002];
vector<pair<int, int>> vec;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	bool allzero = 1;

	cin >> N;
	for (int i = 1; i <= N; i++) {
		int& x = arr[i];
		cin >> x;
		vec.emplace_back(make_pair(x, i));
		if (x) allzero = 0;
	}
	arr[0] = arr[N + 1] = -1;

	if (allzero) {
		cout << 0;
		return 0;
	}

	sort(vec.begin(), vec.end());

	int ans = 0, cur = 1; // cur: 현재 조각수
	int old = -1; // old: 해수면 높이
	for (auto& p : vec) {
		if (old < p.first) ans = max(ans, cur), old = p.first;
		int& pos = p.second;
		cur++;
		if (arr[pos - 1] == -1) cur--;
		if (arr[pos + 1] == -1) cur--;
		arr[pos] = -1;
	}

	cout << ans;
}