BOJ 13308

주유소

문제출처

문제 해석

  • N개의 도시와 M개의 도로가 있을 때, 1에서 N으로 가기 위해 필요한 최소비용을 구하는 문제입니다.
  • 도로를 이동할 때 1km마다 1리터의 기름을 사용합니다.
  • 각 도시에는 주유소가 있으며, 도시마다 주유소의 리터당 가격은 각각 정해져 있습니다.

설계(30)

  • 최소비용을 구해야 하므로 다익스트라를 활용하는 문제인 것 같습니다.
  • 그러나, 어떠한 경우에는 왔던 길로 되돌아와야 최소비용이 되는 경우가 있습니다.
  • 예를 들면, 기존 도시보다 다른 도시의 기름값이 저렴하여, 다른 도시에서 기름을 충전하고 돌아오는 사례입니다.
  • 저는 이를 구현하기 위해서 dist[현재도시][최소기름값]을 이용하였습니다.
  • 현재 최소 기름값과 다음 기름값을 비교하여 최소 기름값을 갱신하도록 하였습니다.

구현(60)

코드
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define MAXN 2501
#define INF 3e10
typedef long long ll;
struct node{
    int cur;
    int cur_w;
    ll cur_wl;
    bool operator < (const node& a) const{
        return a.cur_wl < cur_wl;
    }
};
int Won[MAXN];
vector<pair<int,int> > adj[MAXN];
ll dist[MAXN][MAXN];
int N,M;
ll dijkstra(){
    priority_queue<node> pq;
    pq.push({1,Won[1],0});
    dist[1][Won[1]] = 0;
    while (!pq.empty()){
        int cur = pq.top().cur;
        ll cur_wl = pq.top().cur_wl;
        int cur_w = pq.top().cur_w;
        pq.pop();
        if (cur == N) return cur_wl;
        for (int i = 0 ;i  < adj[cur].size(); i++){
            int next = adj[cur][i].first;
            int next_l = adj[cur][i].second;
            ll next_wl = cur_wl + (ll)cur_w*next_l;
            if (dist[next][min(cur_w,Won[next])] > next_wl){
                dist[next][min(cur_w,Won[next])] = next_wl;
                pq.push({next,min(cur_w,Won[next]),next_wl});
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> N >> M;
    for (int i = 1; i <=N ; i++){
        cin >> Won[i];
    }
    while (M--){
        int u,v,l;
        cin >> u >> v >> l;
        adj[u].push_back({v,l});
        adj[v].push_back({u,l});
    }
    for (int i = 1; i <=N; i++){
        for (int j = 1; j<MAXN; j++){
            dist[i][j] = INF;
        }
    }
    cout << dijkstra() <<"\n";
    return 0;
}
디버깅
  • 처음에 구현했을 때, 런타임에러가 나서 이를 체크하는데 시간이 오래걸렸습니다.
  • 곰곰히 생각해보니 최대로 나올 수 있는 비용은 2500x2500x4000 = 2.5x10^9정도로 int형 범위를 넘어서는 것을 알았습니다.
  • long long형으로 바꾸는데 함수의 리턴 자료형을 바꾸지 않아 이를 찾는데 오래걸렸습니다.
  • 또한, dist[현재도시][현재기름값]으로 구현하고 큐에 최소기름값을 넣는 실수를 하였습니다.
제출결과
  • 328ms

마무리

  • 다익스트라 알고리즘을 응용한 문제입니다. 아이디어는 찾았지만 구현에 시간이 많이 걸린 문제입니다. 자료형을 더욱 신경써서 구현해야 할 것 같습니다.