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;
}