알고리즘 풀이/프로그래머스

[프로그래머스][JAVA] 합승 택시 요금 (그래프, DP, 플로이드 와샬)

배게 2021. 9. 8. 03:13
728x90

index부분 이상하게 넣어서 삽질 너무 많이 함 index는 처음 코딩할 때 2번씩 검사해도 안 부족할듯

 

++ floyd 알고리즘 부분 3중 for문에서 k, i, j 순으로 해줘야함, i, j, k로 했다가 40/100나왔음

경로를 한다는 개념은 알겠는데 3중 for문의 가장 첫번째 for문이 경로 지점을 고정시켜주는 것임을

명확하게 이해를 못한 것 같다

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
import java.util.*;
 
class Solution {
    static final int INF = 400*100000;
    int[][] Dist = new int[200][200];
    
    void floyd(int n){
        for(int k=0; k<n; k++){    
            for(int i=0; i<n; i++){    
                for(int j=0; j<n; j++){    
                    if(Dist[i][k]+Dist[k][j]<Dist[i][j]){    
                        Dist[i][j]=Dist[i][k]+Dist[k][j];    
                    }    
                }    
            }    
        }
        
        
        
    }
    
    
    
    public int solution(int n, int s, int a, int b, int[][] fares) {
        
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                if(i==j){
                    Dist[i][j]=0;
                }else{
                    Dist[i][j]=INF;
                }
            }
        }
        
        for(int[] edge : fares){
            Dist[edge[0]-1][edge[1]-1= edge[2];
            Dist[edge[1]-1][edge[0]-1= edge[2];
            
        }
        
        floyd(n);
        
        
        int answer = INF;
        s--;
        a--;
        b--;
        
        for(int k=0; k<n; k++){
            answer = Math.min(answer, Dist[s][k]+Dist[k][a]+Dist[k][b]);
            
        }
        
        
        return answer;
    }
}
cs