Leetcode 1436.旅行終點站
1 題目描述(Leetcode題目鏈接)
給你一份旅遊線路圖,該線路圖中的旅行線路用數組 paths 表示,其中 paths[i] = [cityAi, cityBi] 表示該線路將會從 cityAi 直接前往 cityBi 。請你找出這次旅行的終點站,即沒有任何可以通往其他城市的線路的城市。
題目數據保證線路圖會形成一條不存在循環的線路,因此只會有一個旅行終點站。
輸入:paths = [["London","New York"],["New York","Lima"],["Lima","Sao Paulo"]]
輸出:"Sao Paulo"
解釋:從 "London" 出發,最後抵達終點站 "Sao Paulo" 。本次旅行的路線是 "London" -> "New York" -> "Lima" -> "Sao Paulo" 。
輸入:paths = [["B","C"],["D","B"],["C","A"]]
輸出:"A"
解釋:所有可能的線路是:
"D" -> "B" -> "C" -> "A".
"B" -> "C" -> "A".
"C" -> "A".
"A".
顯然,旅行終點站是 "A" 。
提示:
- 1 <= paths.length <= 100
- paths[i].length == 2
- 1 <= cityAi.length, cityBi.length <= 10
- cityAi != cityBi
- 所有字符串均由大小寫英文字母和空格字符組成。
2 題解
第一想法是統計一下每個節點的出度,0的那個就是結果。
class Solution:
def destCity(self, paths: List[List[str]]) -> str:
degree = collections.defaultdict(int)
for u, v in paths:
degree[u] += 1
if v not in degree:
degree[v] = 0
for u, i in degree.items():
if i == 0:
return u
用集合的話,就是可到達的節點集合減去出發節點集合。
class Solution:
def destCity(self, paths: List[List[str]]) -> str:
beg = set([city[0] for city in paths])
end = set([city[1] for city in paths])
return (end - beg).pop()