AcWing 693. 行程排序

问题描述

玛丽需要从某地飞往另一目的地，由于没有直达飞机，所以需要在中途转很多航班。
例如：SFO -> DFW DFW -> JFK JFK -> MIA MIA -> ORD。
显然旅途中不可能到同一中转城市两次或以上，因为这没有意义。
不幸的是，她将自己的机票的顺序搞乱了，将机票按乘坐顺序整理好对她来说不是一件容易的事。
请你帮助玛丽整理机票，使机票按正确顺序排列。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。
每组数据第一行包含整数 N。
接下来 2N 行，每 2 行一组，表示一张机票的信息，每行包含一个字符串，其中第一行表示出发地，第二行表示目的地。

输出格式

每组数据输出一个结果，每个结果占一行。
结果表示为 Case #x: y，其中 x 是组别编号（从 1 开始），y 是表示实际行程的机票列表，行程中的每个航段应以 source-destination 的形式输出，航段之间用空格隔开。

数据范围

1≤T≤100,
1≤N≤10000

输入样例：

2
1
SFO
DFW
4
MIA
ORD
DFW
JFK
SFO
DFW
JFK
MIA

输出样例：

Case #1: SFO-DFW
Case #2: SFO-DFW DFW-JFK JFK-MIA MIA-ORD

解题思路

读题发现有点链表模拟数组的感觉。首先要将所有边存储下来，用哈希表next存储每个点走到哪个点，然后将每个有入度的点存入哈希表s中。找头结点时就通过判断哈希表next中每组数据的起点是否在s中，不在则说明是头结点，然后遍历即可。

AC代码

#include <iostream>
#include <unordered_map>
#include <unordered_set>
using namespace std;
int main()
{
    int t;
    cin>>t;
    for(int cases=1;cases<=t;cases++)
    {
        int n;
        cin>>n;
        unordered_map<string,string> next;//哈希表存储每个点可以走到哪个点
        unordered_set<string> s; //存储哪些点有前驱（即哪些点有入度）
        while(n--)
        {
            string a,b;
            cin>>a>>b;
            next[a]=b;
            s.insert(b);
        }
        string head;
        for(auto [a,b]:next) //找头结点
        if(!s.count(a)) //如果a没有前驱结点,那么a就是头结点
        {
            head=a;
            break;
        } 
        printf("Case #%d: ",cases);
        while(next[head].size())
        {
            cout<<head<<'-'<<next[head]<<' ';
            head=next[head];
        }
        cout<<endl;
    }
    return 0;
}