본문 바로가기
Algorithm/프로그래머스풀이

[알고리즘 문제풀이] 프로그래머스 - 합승 택시 요금 / JAVA(자바)

by 계범 2022. 2. 18.

https://programmers.co.kr/learn/courses/30/lessons/72413

 

코딩테스트 연습 - 합승 택시 요금

6 4 6 2 [[4, 1, 10], [3, 5, 24], [5, 6, 2], [3, 1, 41], [5, 1, 24], [4, 6, 50], [2, 4, 66], [2, 3, 22], [1, 6, 25]] 82 7 3 4 1 [[5, 7, 9], [4, 6, 4], [3, 6, 1], [3, 2, 3], [2, 1, 6]] 14 6 4 5 6 [[2,6,6], [6,3,7], [4,6,7], [6,5,11], [2,5,12], [5,3,20], [2,4

programmers.co.kr

/**
    플로이드 워셜 풀이
    
    1. 전체지점에서 전체 지점 이동 최소비용 구하기
    
    2. 시작지점에서 따로 가는 것을 초기값 지정
    
    3. 시작지점에서 중간지점까지 가는 비용 + 중간지점에서 a, 중간지점에서 b 더해서 최소비용 반환
**/

import java.util.*;

class Solution {
    
    public int solution(int n, int s, int a, int b, int[][] fares) {
        int answer = 0;
        
        int maxInt = 100_001*200;
        
        int[][] costs = new int[n+1][n+1];
        
        for(int i = 0; i <= n; i++){
            Arrays.fill(costs[i],maxInt);
            costs[i][i] = 0;
        }
        
        for(int[] fare : fares){
            int start = fare[0];
            int end = fare[1];
            int cost = fare[2];
            
            costs[start][end] = cost;
            costs[end][start] = cost;
        }
        
        for(int mid = 1; mid <= n; mid++){
            for(int i = 1; i <= n; i++){
                for(int j = 1; j <= n; j++){
                    costs[i][j] = Math.min(costs[i][j], costs[i][mid] + costs[mid][j]);
                }
            }
        }
        answer = maxInt;
        
        for(int i = 1; i <=n; i++){
            answer = Math.min(answer, costs[s][i] + costs[i][a] + costs[i][b]);
        }
        
        return answer;
    }
}

댓글