How We Coding

### Dijkstra ###


[1162] 도로포장 : http://boj.kr/1162


<소스코드>


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
 
const int INF = 1e9;
 
typedef pair<intint> P;
typedef pair<int, P> K;
 
int main()
{
    int n, m, k;
    scanf("%d%d%d"&n, &m, &k);
 
    vector<P> g[10001];
    while(m--) {
        int a, b, c;
        scanf("%d%d%d"&a, &b, &c);
        g[a].push_back(P(b, c));
        g[b].push_back(P(a, c));
    }
 
    vector<vector<int> > dist(k+1vector<int>(n+1, INF)); 
    dist[k][1= 0// kth pave, cur vertex
 
    priority_queue<K> pq;
    pq.push(K(0, P(1, k))); // dist, v, k
 
    bool visited[21][10001]={0}; 
  
    while(!pq.empty()) {
        int curV, curK;
        do {
            curV = pq.top().second.first;
            curK = pq.top().second.second; 
            pq.pop();
        } while(!pq.empty() && visited[curK][curV]);
  
        if(visited[curK][curV]) break;
        visited[curK][curV] = 1;
 
        for(auto &p : g[curV]) {
            int nextV = p.first;
            int d = p.second;
            
            if(dist[curK][nextV] > dist[curK][curV] + d) {
                dist[curK][nextV] = dist[curK][curV] + d;
                pq.push(K(-dist[curK][nextV], P(nextV, curK)));
            }
            if(curK-1 >= 0 && dist[curK-1][nextV] > dist[curK][curV]) {
                dist[curK-1][nextV] = dist[curK][curV];
                pq.push(K(-dist[curK-1][nextV], P(nextV, curK-1)));
            }
        }
    }
    
    int ans = 1e9;
    for(int i=0; i<=k; i++) {
        ans = min(ans, dist[i][n]);
    }
    
    printf("%d\n", ans);
    return 0;
}
 
cs

>> dist[k][i] : 포장을 k개 할 수 있는 상태에서 시작정점에서 i번 정점까지의 최단거리, 

    k값의 변화가 없는 겨우, 우리가 지금까지 구했던 다익스트라 dist 배열이다.


'BOJ > Shortest Path' 카테고리의 다른 글

[5917] 거의 최단경로  (0) 2018.07.26
[13424] 비밀 모임  (0) 2018.07.24
[10282] 해킹  (0) 2018.07.24