ps

[백준] 1981 - 배열에서 이동

kariskan 2022. 8. 18. 15:59

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

 

1981번: 배열에서 이동

n×n짜리의 배열이 하나 있다. 이 배열의 (1, 1)에서 (n, n)까지 이동하려고 한다. 이동할 때는 상, 하, 좌, 우의 네 인접한 칸으로만 이동할 수 있다. 이와 같이 이동하다 보면, 배열에서 몇 개의 수를

www.acmicpc.net

 

이 문제와 거의 똑같은 문제이다.

 

배열 하나를 이용해 중복제거 후 원소들을 일렬 오름차순으로 정렬하고 가질 수 있는 최소와 최댓값을 투 포인터로 갱신하며 bfs를 돌리면 끝이다.

 

알고리즘 태그에는 이분 탐색만 존재하는데, 투포인터가 훨씬 구현하기 쉽다.

 

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

int n, map[100][100], visit[100][100];
vector<int> seq;
int x[4] = { 0,0,1,-1 };
int y[4] = { 1,-1,0,0 };

bool bfs(int left, int right) {

	if (map[0][0] < left || map[0][0] > right) return false;

	memset(visit, 0, sizeof(visit));
	queue<pair<int, int>> q;

	q.push({ 0,0 });
	visit[0][0] = 1;

	while (!q.empty()) {
		int a = q.front().first;
		int b = q.front().second;
		q.pop();

		for (int k = 0; k < 4; k++) {
			int nx = a + x[k];
			int ny = b + y[k];
			if (nx >= 0 && nx < n && ny >= 0 && ny < n && !visit[nx][ny] && map[nx][ny] >= left && map[nx][ny] <= right) {
				visit[nx][ny] = 1;
				q.push({ nx,ny });
			}
		}
	}

	return visit[n - 1][n - 1];
}

int main() {

	cin >> n;

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> map[i][j];
			seq.push_back(map[i][j]);
		}
	}

	sort(seq.begin(), seq.end());
	seq.erase(unique(seq.begin(), seq.end()), seq.end());

	int left = 0, right = 0, ans = seq[seq.size() - 1] - seq[0];

	while (left <= right && right < seq.size()) {

		if (bfs(seq[left], seq[right])) {
			ans = min(ans, seq[right] - seq[left]);
			left++;
		}
		else {
			right++;
		}
	}

	cout << ans;
}