Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

สร้างแผนการเดินทางใหม่ใน C++


สมมติว่าเรามีรายการตั๋วเครื่องบินที่แสดงโดยคู่ของสนามบินต้นทางและปลายทาง เช่น [จาก, ถึง] เราต้องสร้างแผนการเดินทางใหม่ตามลำดับ ตั๋วทั้งหมดเป็นของผู้ชายที่เดินทางออกจาก JFK ดังนั้นกำหนดการเดินทางต้องเริ่มต้นด้วย JFK

ดังนั้นหากอินพุตเป็น [["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]] แล้ว เอาต์พุตจะเป็น ["JFK", "MUC", "LHR", "SFO", "SJC"].

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

  • กำหนดอาร์เรย์ ret และแผนที่ที่เรียกว่ากราฟ

  • กำหนดวิธีการที่เรียกว่าการเยี่ยมชม จะใช้ชื่อสนามบินเป็นข้อมูลป้อนเข้า

  • ในขณะที่ขนาดของกราฟ[สนามบิน]ไม่ใช่ 0

    • x :=องค์ประกอบแรกของกราฟ[สนามบิน]

    • ลบองค์ประกอบแรกออกจาก graph[airport]

    • โทรไปเยี่ยม(x)

  • แทรกสนามบินเข้า ret

  • จากวิธีหลัก ให้ทำดังต่อไปนี้ −

  • สำหรับฉันอยู่ในช่วง 0 ถึงขนาดของอาร์เรย์ทิกเกอร์

    • u :=ตั๋ว[i, 0], v :=ตั๋ว[i, 1] แทรก v ลงในกราฟ[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, ]