Advertisement
Alex_St

Untitled

Nov 20th, 2022
44
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 14.12 KB | None | 0 0
  1. #include <cassert>
  2. #include <iostream>
  3. #include <map>
  4. #include <string>
  5. #include <vector>
  6. #include <sstream>
  7.  
  8. using namespace std;
  9. using namespace std::string_literals;
  10.  
  11. enum class QueryType {
  12.     NewBus,
  13.     BusesForStop,
  14.     StopsForBus,
  15.     AllBuses,
  16. };
  17.  
  18. struct Query {
  19.     QueryType type;
  20.     string bus;
  21.     string stop;
  22.     vector<string> stops;
  23. };
  24.  
  25.  
  26. istream& operator>>(istream& in, Query& q)
  27. {
  28.     string temp_s1;
  29.     in >> temp_s1;
  30.     //cout << temp_s1;
  31.     if (temp_s1 == "NewBus"s)
  32.     {
  33.         q.type = QueryType::NewBus;
  34.     }
  35.     else if (temp_s1 == "BusesForStop"s)
  36.     {
  37.         q.type = QueryType::BusesForStop;
  38.         in >> q.stop; //в этом случае в переменную структуры будет записано название остановки, для которой надо найти номера автобуса
  39.         return in;
  40.     }
  41.     else if (temp_s1 == "StopsForBus"s)
  42.     {
  43.         q.type = QueryType::StopsForBus;
  44.         in >> q.bus; //в этом случае в переменную структуры будет записан номер остановки, для которой надо найти номера автобуса
  45.         return in;
  46.     }
  47.     else if (temp_s1 == "AllBuses"s)
  48.     {
  49.         q.type = QueryType::AllBuses; //ничего дополнительно не заполняем в структуре Query
  50.         return in;
  51.     }
  52.     else
  53.     {
  54.         return in;
  55.     }
  56.     in >> q.bus; //в этом случае в переменную структуры будет записан номер автобуса (его название)
  57.  
  58.     int stop_count;
  59.     in >> stop_count;
  60.  
  61.     string temp_s;
  62.     q.stops.clear();
  63.  
  64.     for (size_t i = 0; i < stop_count; ++i)
  65.     {
  66.         in >> temp_s;
  67.         q.stops.push_back(temp_s);
  68.     }
  69.  
  70.     return in;
  71. }
  72.  
  73. struct BusesForStopResponse
  74. {
  75.     vector<string> buses_for_stop;
  76. };
  77.  
  78. ostream& operator<<(ostream& os, const BusesForStopResponse& r)
  79. {
  80.     if (r.buses_for_stop.empty())
  81.     {
  82.         os << "No stop";
  83.         return os;
  84.     }
  85.  
  86.     for (size_t i = 0; i < r.buses_for_stop.size();++i)
  87.     {
  88.         if (i != (r.buses_for_stop.size() - 1))
  89.             os << r.buses_for_stop[i] << " ";
  90.         else
  91.             os << r.buses_for_stop[i];
  92.     }
  93.  
  94.     return os;
  95. }
  96.  
  97. struct StopsForBusResponse
  98. {
  99.     vector<string> stops_for_bus;
  100. };
  101.  
  102. ostream& operator<<(ostream& os, const StopsForBusResponse& r)
  103. {
  104.  
  105.  
  106.     if (r.stops_for_bus.empty())
  107.     {
  108.         os << "No bus";
  109.         return os;
  110.     }
  111.  
  112.     for (size_t i = 0; i < r.stops_for_bus.size();++i)
  113.     {
  114.         if (i != (r.stops_for_bus.size() - 1))
  115.             os << r.stops_for_bus[i] << endl;
  116.         else
  117.             os << r.stops_for_bus[i];
  118.     }
  119.  
  120.     return os;
  121. }
  122.  
  123. struct AllBusesResponse {
  124.     vector<string> all_buses;
  125. };
  126.  
  127. ostream& operator<<(ostream& os, const AllBusesResponse& r)
  128. {
  129.     if (r.all_buses.empty())
  130.         os << "No buses";
  131.  
  132.     for (size_t i = 0; i < r.all_buses.size();++i)
  133.     {
  134.         if (i != (r.all_buses.size() - 1))
  135.             os << r.all_buses[i] << endl;
  136.         else
  137.             os << r.all_buses[i];
  138.     }
  139.  
  140.  
  141.     return os;
  142. }
  143.  
  144. class BusManager {
  145.  
  146. private:
  147.     map<string, vector<string>> buses_to_stops_, stops_to_buses_;
  148.  
  149. public:
  150.     void AddBus(const string& bus, const vector<string>& stops)
  151.     {
  152.         auto& ref = buses_to_stops_[bus];
  153.         for (const auto& stop : stops)
  154.         {
  155.             ref.push_back(stop);
  156.             stops_to_buses_[stop].push_back(bus);
  157.         }
  158.     }
  159.  
  160.     BusesForStopResponse GetBusesForStop(const string& stop) const
  161.     {
  162.         BusesForStopResponse temp_res;
  163.         if (stops_to_buses_.count(stop) == 0)
  164.             return temp_res;
  165.  
  166.         for (const auto& bus : stops_to_buses_.at(stop))
  167.             temp_res.buses_for_stop.push_back(bus);
  168.  
  169.         return temp_res;
  170.     }
  171.  
  172.     StopsForBusResponse GetStopsForBus(const string& bus) const
  173.     {
  174.         StopsForBusResponse temp_res;
  175.         if (buses_to_stops_.count(bus) == 0)
  176.             return temp_res;
  177.  
  178.         for (const auto& stop : buses_to_stops_.at(bus))
  179.         {
  180.             string temp_str = "Stop "s + stop + ":";
  181.             if (stops_to_buses_.at(stop).size() == 1)
  182.             {
  183.                 temp_str += " no interchange"s;
  184.                 temp_res.stops_for_bus.push_back(temp_str);
  185.                 continue;
  186.             }
  187.  
  188.             for (const auto& additional_bus : stops_to_buses_.at(stop))
  189.                 if (additional_bus != bus)
  190.                     temp_str += " "s + additional_bus;
  191.  
  192.             temp_res.stops_for_bus.push_back(temp_str);
  193.         }
  194.  
  195.         return temp_res;
  196.     }
  197.  
  198.     AllBusesResponse GetAllBuses() const
  199.     {
  200.         AllBusesResponse temp_res;
  201.  
  202.         if (buses_to_stops_.empty())
  203.             return temp_res;
  204.  
  205.         for (const auto& bus : buses_to_stops_) //можно наверное переделать в цикл for только по первому в паре, чтобы не перебирать во внешнем цикле второе составляющее пары?
  206.         {
  207.             string temp_str;
  208.            
  209.             for (const auto& stop : buses_to_stops_.at(bus.first))
  210.                 temp_str += " "s + stop;
  211.  
  212.             temp_str = "Bus " + bus.first + ":" + temp_str;
  213.  
  214.             temp_res.all_buses.push_back(temp_str);
  215.         }
  216.  
  217.         return temp_res;
  218.  
  219.     }
  220. };
  221.  
  222. //void TestInput()
  223. //{
  224. //  BusManager test_bm;
  225. //  Query test_subj;
  226. //
  227. //  istringstream input;
  228. //  ostringstream output;
  229. //
  230. //
  231. //  string test_query1 = "AllBuses"s;
  232. //  input.str(test_query1); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  233. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  234. //  output << test_bm.GetAllBuses(); //делаем вывод заданного типа/формата в наш специально созданный поток
  235. //  cout<< test_bm.GetAllBuses();
  236. //  cout << endl;
  237. //  assert(output.str() == "No buses"); //сверяем то что было выведено с помощью output << test_bm.GetAllBuses(); с контрольным результатом
  238. //
  239. //
  240. //  input.str("");
  241. //  input.clear();
  242. //  string test_query2 = "BusesForStop Marushkino"s;
  243. //  input.str(test_query2); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  244. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  245. //  output << test_bm.GetBusesForStop(test_subj.stop); //делаем вывод заданного типа/формата в наш специально созданный поток
  246. //  cout<< test_bm.GetBusesForStop(test_subj.stop);
  247. //  cout << endl;
  248. //  assert(output.str() == "No stop"); //сверяем то что было выведено с помощью output << test_bm.GetBusesForStop(test_subj.stop); с контрольным результатом
  249. //
  250. //
  251. //  input.str("");
  252. //  input.clear();
  253. //  string test_query3 = "StopsForBus 32K"s;
  254. //  input.str(test_query3); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  255. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  256. //  output << test_bm.GetStopsForBus(test_subj.bus); //делаем вывод заданного типа/формата в наш специально созданный поток
  257. //  cout<< test_bm.GetStopsForBus(test_subj.bus);
  258. //  assert(output.str() == "No bus"); //сверяем то что было выведено с помощью output << test_bm.GetStopsForBus(test_subj.bus) с контрольным результатом
  259. // 
  260. //
  261. //  input.str("");
  262. //  input.clear();
  263. //  string test_query4 = "NewBus 32 3 Tolstopaltsevo Marushkino Vnukovo"s;
  264. //  input.str(test_query4); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  265. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  266. //  test_bm.AddBus(test_subj.bus, test_subj.stops); //передаем в тестовый экземпляр класса данные на основе заполненной ранее в input >> test_subj; структуры
  267. //
  268. //
  269. //  input.str("");
  270. //  input.clear();
  271. //  string test_query5 = "NewBus 32K 6 Tolstopaltsevo Marushkino Vnukovo Peredelkino Solntsevo Skolkovo"s;
  272. //  input.str(test_query5); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  273. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  274. //  test_bm.AddBus(test_subj.bus, test_subj.stops); //передаем в тестовый экземпляр класса данные на основе заполненной ранее в input >> test_subj; структуры
  275. //
  276. //
  277. //  input.str("");
  278. //  input.clear();
  279. //  string test_query6 = "BusesForStop Vnukovo"s;
  280. //  input.str(test_query6); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  281. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  282. //  output << test_bm.GetBusesForStop(test_subj.stop); //делаем вывод заданного типа/формата в наш специально созданный поток
  283. //  cout<< test_bm.GetBusesForStop(test_subj.stop);
  284. //  cout << endl;
  285. //  assert(output.str() == "32 32K"); //сверяем то что было выведено с помощью output << test_bm.GetBusesForStop(test_subj.stop) с контрольным результатом
  286. // 
  287. //
  288. //  input.str("");
  289. //  input.clear();
  290. //  string test_query7 = "NewBus 950 6 Kokoshkino Marushkino Vnukovo Peredelkino Solntsevo Troparyovo"s;
  291. //  input.str(test_query7); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  292. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  293. //  test_bm.AddBus(test_subj.bus, test_subj.stops); //передаем в тестовый экземпляр класса данные на основе заполненной ранее в input >> test_subj; структуры
  294. //
  295. //
  296. //  input.str("");
  297. //  input.clear();
  298. //  string test_query8 = "NewBus 272 4 Vnukovo Moskovsky Rumyantsevo Troparyovo"s;
  299. //  input.str(test_query8); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  300. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  301. //  test_bm.AddBus(test_subj.bus, test_subj.stops); //передаем в тестовый экземпляр класса данные на основе заполненной ранее в input >> test_subj; структуры
  302. //
  303. //
  304. //  input.str("");
  305. //  input.clear();
  306. //  string test_query9 = "StopsForBus 272"s;
  307. //  input.str(test_query9); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  308. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  309. //  output << test_bm.GetStopsForBus(test_subj.bus); //делаем вывод заданного типа/формата в наш специально созданный поток
  310. //  cout<< test_bm.GetStopsForBus(test_subj.bus);
  311. ////    assert(output.str() == R"(Stop Vnukovo : 32 32K 950
  312. ////Stop Moskovsky : no interchange
  313. ////Stop Rumyantsevo : no interchange
  314. ////Stop Troparyovo : 950
  315. ////)"); //сверяем то что было выведено с помощью output << test_bm.GetStopsForBus(test_subj.bus) с контрольным результатом
  316. // 
  317. //     
  318. //  input.str("");
  319. //  input.clear();
  320. //  string test_query10 = "AllBuses"s;
  321. //  input.str(test_query10); //заводим в поток заранее созданную строку, имитируя ввод с клавиатуры
  322. //  input >> test_subj; //вводим в заготовленную переменную данные, которые были введены в поток используя input.str ранее
  323. //  output << test_bm.GetAllBuses(); //делаем вывод заданного типа/формата в наш специально созданный поток
  324. //  cout << test_bm.GetAllBuses();
  325. ////    assert(output.str() == R"(Bus 272: Vnukovo Moskovsky Rumyantsevo Troparyovo
  326. ////Bus 32 : Tolstopaltsevo Marushkino Vnukovo
  327. ////Bus 32K : Tolstopaltsevo Marushkino Vnukovo Peredelkino Solntsevo Skolkovo
  328. ////Bus 950 : Kokoshkino Marushkino Vnukovo Peredelkino Solntsevo Troparyovo
  329. ////)"); //сверяем то что было выведено с помощью output << test_bm.GetAllBuses(); с контрольным результатом
  330. //
  331. //}
  332.  
  333.  
  334. // Реализуйте функции и классы, объявленные выше, чтобы эта функция main
  335. // решала задачу "Автобусные остановки"
  336.  
  337. int main() {
  338.    
  339.     //TestInput();
  340.    
  341.     int query_count;
  342.     Query q;
  343.  
  344.     cin >> query_count;
  345.  
  346.     BusManager bm;
  347.     for (int i = 0; i < query_count; ++i) {
  348.         cin >> q;
  349.         switch (q.type) {
  350.             case QueryType::NewBus:
  351.                 bm.AddBus(q.bus, q.stops);
  352.                 break;
  353.             case QueryType::BusesForStop:
  354.                 cout << bm.GetBusesForStop(q.stop) << endl;
  355.                 break;
  356.             case QueryType::StopsForBus:
  357.                 cout << bm.GetStopsForBus(q.bus) << endl;
  358.                 break;
  359.             case QueryType::AllBuses:
  360.                 cout << bm.GetAllBuses() << endl;
  361.                 break;
  362.         }
  363.     }
  364. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement