LeetCode刷题笔记 标准模板库巧解算法题 多重集合

332 重新安排行程

给定一个人坐过的一些飞机的起止机场,已知这个人从 JFK 起飞,那么这个人是按什么顺序飞的;如果存在多种可能性,返回字母序最小的那种。

输入是一个二维字符串数组,表示多个起止机场对子;输出是一个一维字符串数组,表示飞行顺序。

输入:tickets = [[“JFK”,“SFO”],[“JFK”,“ATL”],[“SFO”,“ATL”],[“ATL”,“JFK”],[“ATL”,“SFO”]]
输出:[“JFK”,“ATL”,“JFK”,“SFO”,“ATL”,“SFO”]
解释:另一种有效的行程是 [“JFK”,“SFO”,“ATL”,“JFK”,“ATL”,“SFO”] ,但是它字典排序更大更靠后。

解析:

​ 本题可以使用哈希表记录机票的起飞机场和落地机场,由于可能存在多个不同落地机场,所以需要使用集合来存储。又由于可能存在重复路线,所以要保存落地机场的重复值,且要求结果的返回顺序是按照最小字母序,刚好可以使用多重集合 multiset,不仅可以存储重复值,而且多重集合的默认元素保存顺序是递增的,例如"SFO","ATL","JFK" 在多重集合中的保存顺序是 "ATL","JFK","SFO"。上例子的哈希表保存结果为:

Hash_key Hash_value
JFK {ATL, SFO}
SFO {ATL}
ATL {JFK, SFO}

​ 保存好起飞机场和落地机场的关系后,我们使用栈来恢复一条完整的从最终落地机场到达起始机场的路线。从起始机场开始,按照转移顺序将机场逐个压入栈。取当前栈顶作为下一个落地机场,如果栈顶为空则说明到达一个终点,将其加入结果。如果不为空,则将其作为起飞机场对应的第一个落地机场入栈并移除。完成上述过程就恢复出了一条从终点到起点的飞行路线。上例恢复的路线为JFK->ATL->JFK->SFO->ATL->SFO

class Solution {
    
    
public:
    vector<string> findItinerary(vector<vector<string>>& tickets) {
    
    
        // 构建哈希表多重集合,保存起飞机场对应的落地机场
        unordered_map<string, multiset<string>> hash;
        for(const auto ticket: tickets){
    
    
            hash[ticket[0]].insert(ticket[1]);
        }
        // 使用栈恢复从终点到达起点的逆向路线
        vector<string> ans;
        stack<string> st;
        st.push("JFK");
        while(!st.empty()){
    
    
            string end = st.top();
            // 当前机场对应的落地机场为空,说明他是一个终点
            if(hash[end].empty()){
    
    
                ans.push_back(end);
                st.pop();
            }else{
    
    
                st.push(*hash[end].begin());
                hash[end].erase(hash[end].begin());
            }
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
};

参考资料

LeetCode 101:和你一起轻松刷题(C++) 第 11 章 妙用数据结构

Guess you like

Origin blog.csdn.net/qq_41773806/article/details/120862624