Merge branch 'master' of https://github.com/TrainingByPackt/CPP-Data-Structures-and-Algorithms

Author: aniketshedge-packt

Lesson02/Exercise06/6.cpp

@@ -0,0 +1,101 @@
+#include <iostream>
+#include <vector>
+#include <algorithm>
+
+enum class city: int
+{
+	MOSCOW,
+	LONDON,
+	ISTANBUL,
+	SEATTLE,
+	DUBAI,
+	MUMBAI,
+	SINGAPORE
+};
+
+std::ostream& operator<<(std::ostream& os, const city c)
+{
+	switch(c)
+	{
+		case city::MOSCOW:
+			os << "MOSCOW";
+			return os;
+		case city::LONDON:
+			os << "LONDON";
+			return os;
+		case city::ISTANBUL:
+			os << "ISTANBUL";
+			return os;
+		case city::SEATTLE:
+			os << "SEATTLE";
+			return os;
+		case city::DUBAI:
+			os << "DUBAI";
+			return os;
+		case city::MUMBAI:
+			os << "MUMBAI";
+			return os;
+		case city::SINGAPORE:
+			os << "SINGAPORE";
+			return os;
+		default:
+			return os;
+	}
+}
+
+struct graph
+{
+	std::vector<std::vector<std::pair<int, int>>> data;
+
+graph(int n)
+{
+	data = std::vector(n, std::vector<std::pair<int, int>>());
+}
+
+void addEdge(const city c1, const city c2, int dis)
+{
+	std::cout << "ADD: " << c1 << "-" << c2 << "=" << dis << std::endl;
+
+	auto n1 = static_cast<int>(c1);
+	auto n2 = static_cast<int>(c2);
+	data[n1].push_back({n2, dis});
+	data[n2].push_back({n1, dis});
+}
+
+void removeEdge(const city c1, const city c2)
+{
+	std::cout << "REMOVE: " << c1 << "-" << c2 << std::endl;
+
+	auto n1 = static_cast<int>(c1);
+	auto n2 = static_cast<int>(c2);
+	std::remove_if(data[n1].begin(), data[n1].end(), [n2](const auto& pair)
+		{
+			return pair.first == n2;
+		});
+	std::remove_if(data[n2].begin(), data[n2].end(), [n1](const auto& pair)
+		{
+			return pair.first == n1;
+		});
+}
+};
+
+int main()
+{
+	graph g(7);
+	g.addEdge(city::LONDON, city::MOSCOW, 900);
+	g.addEdge(city::LONDON, city::ISTANBUL, 500);
+	g.addEdge(city::LONDON, city::DUBAI, 1000);
+	g.addEdge(city::ISTANBUL, city::MOSCOW, 1000);
+	g.addEdge(city::ISTANBUL, city::DUBAI, 500);
+	g.addEdge(city::DUBAI, city::MUMBAI, 200);
+	g.addEdge(city::ISTANBUL, city::SEATTLE, 1500);
+	g.addEdge(city::DUBAI, city::SINGAPORE, 500);
+	g.addEdge(city::MOSCOW, city::SEATTLE, 1000);
+	g.addEdge(city::MUMBAI, city::SINGAPORE, 300);
+	g.addEdge(city::SEATTLE, city::SINGAPORE, 700);
+
+	g.addEdge(city::SEATTLE, city::LONDON, 1800);
+	g.removeEdge(city::SEATTLE, city::LONDON);
+
+	return 0;
+}