LeetCode 1436.旅行终点站

题目描述

给你一份旅游线路图，该线路图中的旅行线路用数组 paths 表示，其中 paths[i] = [cityAi, cityBi] 表示该线路将会从 cityAi 直接前往 cityBi 。请你找出这次旅行的终点站，即没有任何可以通往其他城市的线路的城市。

题目数据保证线路图会形成一条不存在循环的线路，因此恰有一个旅行终点站。

示例 1：

输入：paths = [["London","New York"],["New York","Lima"],["Lima","Sao Paulo"]]
输出："Sao Paulo" 
解释：从 "London" 出发，最后抵达终点站 "Sao Paulo" 。本次旅行的路线是 "London" -> "New York" -> "Lima" -> "Sao Paulo" 。

示例 2：

输入：paths = [["B","C"],["D","B"],["C","A"]]
输出："A"
解释：所有可能的线路是：
"D" -> "B" -> "C" -> "A". 
"B" -> "C" -> "A". 
"C" -> "A". 
"A". 
显然，旅行终点站是 "A" 。

示例 3：

输入：paths = [["A","Z"]]
输出："Z"

提示：

• 1 <= paths.length <= 100
• paths[i].length == 2
• 1 <= cityAi.length,cityBi.length <= 10
• cityAi != cityBi
• 所有字符串均由大小写英文字母和空格字符组成。

思路

本题笔者使用的是哈希表法。首先将每个线路都存入哈希表中，key对应起点，value对应终点。再从任意一个线路开始，依据其终点value来在哈希表中寻找对应的起点，找到了就继续往下找。当找不到对应起点的时候，说明当前线路的终点就是最后的旅行终点站。

代码

C++版：

class Solution {
public:
    string destCity(vector<vector<string>>& paths) {
        // 哈希表法
        // 将每个线路存入哈希表中，key为起点，value为终点
        // 从任意一个线路开始，依据其终点来找对应的起点
        // 当找不到的时候就返回其value，此时value就是终点站
        unordered_map<string,string> strMap(paths.size());
        string next=paths[0][1]; // 从第一个线路开始
        for(int i=0;i<paths.size();i++){
            strMap.insert({paths[i][0],paths[i][1]});
        }
        while(strMap.find(next)!= strMap.end()){
            auto it=strMap.find(next);
            next=it->second;
        }
        return next;

    }
};

Python版：

class Solution:
    def destCity(self, paths: List[List[str]]) -> str:
        str_map = {}
        next_city = paths[0][1] # 从第一个线路开始 
        for path in paths:  
            str_map[path[0]] = path[1]
        while next_city in str_map:  
            next_city = str_map[next_city]
        return next_city