用 C++ 重構行程


假設我們有一張航空機票列表,用出發機場和到達機場對錶示,例如[出發地,目的地],我們必須按順序重構行程。所有機票都屬於一位從 JFK 出發的男士。因此,行程必須以 JFK 開始。

所以,如果輸入類似於[["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]],那麼輸出將是["JFK", "MUC", "LHR", "SFO", "SJC"]。

為了解決這個問題,我們將遵循以下步驟:

  • 定義陣列 ret 和一個名為 graph 的對映。

  • 定義一個名為 visit 的方法。這將以機場名稱作為輸入。

  • 當 graph[機場] 的大小不為 0 時

    • x := graph[機場] 的第一個元素

    • 從 graph[機場] 中刪除第一個元素

    • 呼叫 visit(x)

  • 將機場插入 ret

  • 現在,從主方法開始,執行以下操作:

  • 對於 i 在 0 到 tickers 陣列大小的範圍內

    • u := tickets[i, 0],v := tickets[i, 1],將 v 插入 graph[u]

  • 訪問“JFK”,因為這是第一個機場

  • 反轉列表 ret 並返回

示例 (C++)

讓我們看看下面的實現,以便更好地理解:

 線上演示

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
class Solution {
   public:
   vector <string> ret;
   map < string, multiset <string> > graph;
   vector<string> findItinerary(vector<vector<string>>& tickets) {
      for(int i = 0; i < tickets.size(); i++){
         string u = tickets[i][0];
         string v = tickets[i][1];
         graph[u].insert(v);
      }
      visit("JFK");
      reverse(ret.begin(), ret.end());
      return ret;
   }
   void visit(string airport){
      while(graph[airport].size()){
         string x = *(graph[airport].begin());
         graph[airport].erase(graph[airport].begin());
         visit(x);
      }
      ret.push_back(airport);
   }
};
main(){
   Solution ob;
   vector<vector<string>> v = {{"MUC","LHR"},{"JFK","MUC"},{"SFO","SJC"},{"LHR","SFO"}};
   print_vector(ob.findItinerary(v));
}

輸入

[["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]

輸出

[JFK, MUC, LHR, SFO, SJC, ]

更新於:2020年5月2日

202 次瀏覽

啟動你的職業生涯

透過完成課程獲得認證

開始
廣告
© . All rights reserved.