旅行终点站

2024-10-08

题目：

给你一份旅游线路图，该线路图中的旅行线路用数组 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
  • 所有字符串均由大小写英文字母和空格字符组成。

思路：

由于题目已经做好了限制：只有一个旅行终点站。
如果我们已构造图的思维去看，那么终点站一定没有出边。
对于 paths[i] = [cityAi, cityBi]
我们以set去分别存储 cityA, cityB 和 cityA，也就是存储所有节点 和 有出边的节点
那么剩下的就是终点站

代码：

class Solution {
public:
    string destCity(vector<vector<string>>& paths) {
        set<string> graph;
        set<string> all;
        for (int i = 0; i < paths.size(); i++) {
            graph.insert(paths[i][0]);
            all.insert(paths[i][0]);
            all.insert(paths[i][1]);
        }
        for (string city : all) {
            if (graph.count(city)) continue;
            return city;
        }
        return "";
    }
};