BOJ)1326 폴짝폴짝

문제: icpc.me/1326

각 징검다리에 숫자가 써있고 해당 징검다리에서는 숫자의 배수만큼 떨어져 있는 징검다리까지 점프 할 수 있을때 a에서 b까지 의 최단거리를 출력하는 문제이다.

해당 징검다리에서 갈 수 있는 모든 징검다리까지 cost가 1인 간선을 연결해주고 다익스트라를 이용하여 최단거리를 구해줄 수 있다. 이 때 주의할 점은 배수 이기 때문에 나보다 작은 지점으로도 점프할 수 있다.

#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
int n, a[10001], s, e, dp[10001];
vector<vector<int>> vt;
int main() {
    scanf("%d", &n);
    vt.resize(n + 1);
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    scanf("%d%d", &s, &e);
    memset(dp, -1, sizeof(dp));
    for (int i = 1; i <= n; i++) {
        for (int j = i + a[i]; j <= n; j += a[i])
            vt[i].push_back(j);
        for (int j = i - a[i]; j > 0; j -= a[i])
            vt[i].push_back(j);
    }
    priority_queue<pair<int, int>> pq;
    pq.push({ 0,s });
    while (pq.size()) {
        int here = pq.top().second;
        int cost = -pq.top().first;
        pq.pop();
        if (dp[here] != -1)continue;
        dp[here] = cost;
        for (int there : vt[here]) {
            if (dp[there] != -1)continue;
            pq.push({ -1 - cost,there });
        }
    }
    printf("%d\n", dp[e]);
    return 0;
}