짱아의 개발 기록장

2021 KAKAO BLIND RECRUITMENT : 합승 택시 요금(c++) / 플로이드 와샬 본문

Algorithm/카카오 기출

2021 KAKAO BLIND RECRUITMENT : 합승 택시 요금(c++) / 플로이드 와샬

jungahshin 2021. 3. 9. 22:40
반응형

설명 추가 예정

 

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
#include <string>
#include <vector>
#include <climits>
#include <iostream>
 
using namespace std;
 
int dp[201][201= {0, };
 
int solution(int n, int s, int a, int b, vector<vector<int>> fares) {
    int answer = 1e9;
    
    // 왜 INT_MAX로 하면 안되는거지??? 에러생김...ㅎㅎ
    // 두 개를 더하면 int범위를 초과하게 되기 때문에...쓰레기값이 들어간다..!!
    // 따라서, 적당한 큰 값을 넣어주는 것이 중요하다.
    for(int i=1; i<=n; i++){
        for(int j=1; j<=n; j++){
            dp[i][j] = 20000000;
        }
    }
    
    for(int i=1; i<=n; i++){
        dp[i][i] = 0;
    }
    
    for(int i=0; i<fares.size(); i++){
        int from = fares[i][0];
        int to = fares[i][1];
        int cost = fares[i][2];
                
        dp[to][from] = cost;
        dp[from][to] = cost;
    }
    
    // 거쳐가는 점
    for(int i=1; i<=n; i++){
        // 출발점
        for(int j=1; j<=n; j++){
            // 도착점
            for(int k=1; k<=n; k++){
                dp[j][k] = min(dp[j][k], dp[j][i]+dp[i][k]);
                // dp[i][k] = min(dp[i][k], dp[i][j]+dp[j][k]);
            }
        }
    }
    
    for(int i=1; i<=n; i++){
        answer = min(answer, dp[s][i]+dp[i][a]+dp[i][b]);
    }
    
    return answer;
}
cs
반응형
Comments