본문 바로가기
Problem Solving/카카오 코딩테스트

[카카오 기출] 합승 택시 요금

by ggyongi 2021. 9. 2.
반응형

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

import sys
inf = sys.maxsize
def solution(n, s, a, b, fares):
    table = [[inf]*n for _ in range(n)]
    for i in range(n):
        table[i][i] = 0
        
    for i in range(len(fares)):
        c, d, cost = fares[i]
        table[c-1][d-1] = cost
        table[d-1][c-1] = cost
        
    for i in range(n):
        for j in range(n):
            for k in range(n):
                table[j][k] = min(table[j][k], table[j][i]+ table[i][k])

    answer = 0   
    # not together
    answer = table[a-1][s-1] + table[b-1][s-1]
    
    for i in range(n):     
        cur_cost = table[s-1][i] + table[i][a-1] + table[i][b-1]
        answer = min(answer, cur_cost)
        
    return answer
 

비전공자 네카라 신입 취업 노하우

시행착오 끝에 얻어낸 취업 노하우가 모두 담긴 전자책!

kmong.com

댓글