From 6e6e9c54d93dfa3546891b218c2e183abc5627a6 Mon Sep 17 00:00:00 2001 From: Seongbeom Park Date: Sun, 28 Apr 2024 16:05:51 +0900 Subject: [PATCH] =?UTF-8?q?solve=20=ED=95=A9=EC=8A=B9=20=ED=83=9D=EC=8B=9C?= =?UTF-8?q?=20=EC=9A=94=EA=B8=88?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- run.py | 1 + 합승택시요금/README.md | 61 ++++++++++++++++++++++++++++++++++++++++ 합승택시요금/solution.py | 32 +++++++++++++++++++++ 합승택시요금/test.py | 12 ++++++++ 4 files changed, 106 insertions(+) create mode 100644 합승택시요금/README.md create mode 100644 합승택시요금/solution.py create mode 100644 합승택시요금/test.py diff --git a/run.py b/run.py index 37548e8..a47ec8c 100644 --- a/run.py +++ b/run.py @@ -10,6 +10,7 @@ problems = [ '혼자놀기의달인', '디스크컨트롤러', '점찍기', + '합승택시요금', ] if __name__ == '__main__': diff --git a/합승택시요금/README.md b/합승택시요금/README.md new file mode 100644 index 0000000..d70750a --- /dev/null +++ b/합승택시요금/README.md @@ -0,0 +1,61 @@ +# [합승 택시 요금](https://school.programmers.co.kr/learn/courses/30/lessons/72413) + +## 문제 설명 +[본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.] + +밤늦게 귀가할 때 안전을 위해 항상 택시를 이용하던 무지는 최근 야근이 잦아져 택시를 더 많이 이용하게 되어 택시비를 아낄 수 있는 방법을 고민하고 있습니다. "무지"는 자신이 택시를 이용할 때 동료인 어피치 역시 자신과 비슷한 방향으로 가는 택시를 종종 이용하는 것을 알게 되었습니다. "무지"는 "어피치"와 귀가 방향이 비슷하여 택시 합승을 적절히 이용하면 택시요금을 얼마나 아낄 수 있을 지 계산해 보고 "어피치"에게 합승을 제안해 보려고 합니다. + +2021_kakao_taxi_01.png + +위 예시 그림은 택시가 이동 가능한 반경에 있는 6개 지점 사이의 이동 가능한 택시노선과 예상요금을 보여주고 있습니다. +그림에서 A와 B 두 사람은 출발지점인 4번 지점에서 출발해서 택시를 타고 귀가하려고 합니다. A의 집은 6번 지점에 있으며 B의 집은 2번 지점에 있고 두 사람이 모두 귀가하는 데 소요되는 예상 최저 택시요금이 얼마인 지 계산하려고 합니다. + +그림의 원은 지점을 나타내며 원 안의 숫자는 지점 번호를 나타냅니다. +지점이 n개일 때, 지점 번호는 1부터 n까지 사용됩니다. +지점 간에 택시가 이동할 수 있는 경로를 간선이라 하며, 간선에 표시된 숫자는 두 지점 사이의 예상 택시요금을 나타냅니다. +간선은 편의 상 직선으로 표시되어 있습니다. +위 그림 예시에서, 4번 지점에서 1번 지점으로(4→1) 가거나, 1번 지점에서 4번 지점으로(1→4) 갈 때 예상 택시요금은 10원으로 동일하며 이동 방향에 따라 달라지지 않습니다. +예상되는 최저 택시요금은 다음과 같이 계산됩니다. +4→1→5 : A, B가 합승하여 택시를 이용합니다. 예상 택시요금은 10 + 24 = 34원 입니다. +5→6 : A가 혼자 택시를 이용합니다. 예상 택시요금은 2원 입니다. +5→3→2 : B가 혼자 택시를 이용합니다. 예상 택시요금은 24 + 22 = 46원 입니다. +A, B 모두 귀가 완료까지 예상되는 최저 택시요금은 34 + 2 + 46 = 82원 입니다. + +## [문제] +지점의 개수 n, 출발지점을 나타내는 s, A의 도착지점을 나타내는 a, B의 도착지점을 나타내는 b, 지점 사이의 예상 택시요금을 나타내는 fares가 매개변수로 주어집니다. 이때, A, B 두 사람이 s에서 출발해서 각각의 도착 지점까지 택시를 타고 간다고 가정할 때, 최저 예상 택시요금을 계산해서 return 하도록 solution 함수를 완성해 주세요. +만약, 아예 합승을 하지 않고 각자 이동하는 경우의 예상 택시요금이 더 낮다면, 합승을 하지 않아도 됩니다. + +## [제한사항] +지점갯수 n은 3 이상 200 이하인 자연수입니다. +지점 s, a, b는 1 이상 n 이하인 자연수이며, 각기 서로 다른 값입니다. +즉, 출발지점, A의 도착지점, B의 도착지점은 서로 겹치지 않습니다. +fares는 2차원 정수 배열입니다. +fares 배열의 크기는 2 이상 n x (n-1) / 2 이하입니다. +예를들어, n = 6이라면 fares 배열의 크기는 2 이상 15 이하입니다. (6 x 5 / 2 = 15) +fares 배열의 각 행은 [c, d, f] 형태입니다. +c지점과 d지점 사이의 예상 택시요금이 f원이라는 뜻입니다. +지점 c, d는 1 이상 n 이하인 자연수이며, 각기 서로 다른 값입니다. +요금 f는 1 이상 100,000 이하인 자연수입니다. +fares 배열에 두 지점 간 예상 택시요금은 1개만 주어집니다. 즉, [c, d, f]가 있다면 [d, c, f]는 주어지지 않습니다. +출발지점 s에서 도착지점 a와 b로 가는 경로가 존재하는 경우만 입력으로 주어집니다. + +## [입출력 예] +n s a b fares result +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,8], [4,3,9]] 18 + +## 입출력 예에 대한 설명 +### 입출력 예 #1 +문제 예시와 같습니다. + +### 입출력 예 #2 +2021_kakao_taxi_02.png + +합승을 하지 않고, B는 3→2→1, A는 3→6→4 경로로 각자 택시를 타고 가는 것이 최저 예상 택시요금입니다. +따라서 최저 예상 택시요금은 (3 + 6) + (1 + 4) = 14원 입니다. +입출력 예 #3 +2021_kakao_taxi_03.png + +A와 B가 4→6 구간을 합승하고 B가 6번 지점에서 내린 후, A가6→5` 구간을 혼자 타고 가는 것이 최저 예상 택시요금입니다. +따라서 최저 예상 택시요금은 7 + 11 = 18원 입니다. diff --git a/합승택시요금/solution.py b/합승택시요금/solution.py new file mode 100644 index 0000000..c056e9f --- /dev/null +++ b/합승택시요금/solution.py @@ -0,0 +1,32 @@ +def solution(n, s, a, b, fares): + nodes = {i+1: {} for i in range(n)} + for n1, n2, c in fares: + if n2 in nodes[n1]: + nodes[n1][n2] = min(nodes[n1][n2], c) + nodes[n2][n1] = min(nodes[n1][n2], c) + else: + nodes[n1][n2] = c + nodes[n2][n1] = c + + memo_s = shortest_path(s, nodes) + memo_a = shortest_path(a, nodes) + memo_b = shortest_path(b, nodes) + + answer = float('inf') + for t in range(1, n+1): + if t in memo_s: + answer = min(answer, memo_s[t] + memo_a[t] + memo_b[t]) + return answer + +def shortest_path(x, nodes): + memo = {i+1: float('inf') for i in range(len(nodes))} + memo[x] = 0 + visit = [x] + while len(visit) > 0: + curr = visit.pop(0) + for n, c in nodes[curr].items(): + total_cost = memo[curr] + c + if total_cost < memo[n]: + memo[n] = total_cost + visit += [n] + return memo diff --git a/합승택시요금/test.py b/합승택시요금/test.py new file mode 100644 index 0000000..3ce62df --- /dev/null +++ b/합승택시요금/test.py @@ -0,0 +1,12 @@ +from solution import solution + +test_cases = [ + [[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,8], [4,3,9]]], 18], +] + +for i, o in test_cases: + print('input', *i, 'output', o) + print('solution', solution(*i)) + print()