kolonizace
This commit is contained in:
parent
9bfb15a529
commit
a1a0bc1a2f
|
|
@ -0,0 +1,192 @@
|
|||
#ifndef __PROGTEST__
|
||||
#include <cstdio>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <cctype>
|
||||
#include <cassert>
|
||||
#include <iostream>
|
||||
#include <iomanip>
|
||||
#include <sstream>
|
||||
#include <stdexcept>
|
||||
#include <string>
|
||||
#include <list>
|
||||
#include <vector>
|
||||
#include <stack>
|
||||
#include <queue>
|
||||
#include <deque>
|
||||
#include <map>
|
||||
#include <unordered_map>
|
||||
#include <set>
|
||||
#include <unordered_set>
|
||||
#include <algorithm>
|
||||
#include <memory>
|
||||
using namespace std;
|
||||
|
||||
class Cbase {
|
||||
public:
|
||||
Cbase(string sector, string species, int when) : m_Sector(std::move(sector)), m_Species(std::move(species)), m_When(when) {}
|
||||
~Cbase() = default;
|
||||
|
||||
string m_Sector;
|
||||
string m_Species;
|
||||
int m_When;
|
||||
};
|
||||
|
||||
#endif /* __PROGTEST__ */
|
||||
|
||||
|
||||
|
||||
|
||||
class CUniverse {
|
||||
public:
|
||||
using Race = std::string;
|
||||
using Sector = std::string;
|
||||
using Time = int;
|
||||
|
||||
CUniverse & addConnection(const string & from, const string & to) {
|
||||
_s[from].push_back(to);
|
||||
_s[to].push_back(from);
|
||||
return *this;
|
||||
}
|
||||
|
||||
CUniverse & optimize() {
|
||||
return *this;
|
||||
}
|
||||
|
||||
map<string, string> colonise(const vector<Cbase> & bases) const {
|
||||
std::queue< std::tuple< Sector, Time, Race > > q;
|
||||
std::map< Sector, std::tuple< Time, Race, bool > > d;
|
||||
|
||||
for (auto& i : bases) {
|
||||
q.emplace(i.m_Sector, i.m_When, i.m_Species);
|
||||
d[i.m_Sector] = {i.m_When, i.m_Species, true};
|
||||
}
|
||||
|
||||
while (q.size()) {
|
||||
auto i = q.front();
|
||||
q.pop();
|
||||
if (d.at( std::get<0>(i) ) == std::tuple< Time, Race, bool >(0, "", true)) continue;
|
||||
|
||||
auto it = _s.find(std::get<0>(i));
|
||||
if (it == _s.end()) continue;
|
||||
|
||||
for (const Sector& adj : it->second) {
|
||||
auto itd = d.find(adj);
|
||||
if (itd == d.end()) { // nenalezena
|
||||
q.emplace( adj, std::get<1>(i) + 1, std::get<2>(i) );
|
||||
d[adj] = { std::get<1>(i) + 1, std::get<2>(i), false};
|
||||
continue;
|
||||
}
|
||||
|
||||
//nalezena
|
||||
if (std::get<2>(itd->second)) continue; // startovaci planeta
|
||||
|
||||
if (std::get<0>(itd->second) == std::get<1>(i) + 1) { // srazka
|
||||
d[adj] = {0, "", true};
|
||||
continue;
|
||||
}
|
||||
|
||||
if (std::get<0>(itd->second) > std::get<1>(i) + 1) {
|
||||
q.emplace( adj, std::get<1>(i) + 1, std::get<2>(i) );
|
||||
d[adj] = { std::get<1>(i) + 1, std::get<2>(i), false};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
map<string, string> ret;
|
||||
for (const auto& i : d) {
|
||||
if (i.second == std::tuple< Time, Race, bool >(0, "", true)) continue;
|
||||
ret[i.first] = std::get<1>(i.second);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
private:
|
||||
std::map< Sector, std::vector< Sector > > _s;
|
||||
|
||||
};
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#ifndef __PROGTEST__
|
||||
|
||||
int main ( void ) {
|
||||
CUniverse u1;
|
||||
|
||||
u1 .addConnection("Earth", "X1")
|
||||
.addConnection("Earth", "Y1")
|
||||
.addConnection("Vulcan", "X1")
|
||||
.addConnection("Vulcan", "Y2")
|
||||
.addConnection("Y2", "Y1")
|
||||
.addConnection("Kronos", "X1")
|
||||
.addConnection("X1", "X2")
|
||||
.addConnection("X2", "X3")
|
||||
.optimize();
|
||||
|
||||
auto r1 = u1.colonise({{"Earth", "Humans", 0}, {"Vulcan", "Vulcans", 0}, {"Kronos", "Clingons", 0}});
|
||||
assert(r1 == (map<string, string> ({{"Earth", "Humans"}, {"Y1", "Humans"}, {"Vulcan", "Vulcans"}, {"Y2", "Vulcans"}, {"Kronos", "Clingons"}})));
|
||||
auto r2 = u1.colonise({{"Earth", "Humans", 0}, {"Vulcan", "Vulcans", 0}, {"Kronos", "Humans", 0}});
|
||||
assert(r2 == (map<string, string> ({{"Earth", "Humans"}, {"Y1", "Humans"}, {"Vulcan", "Vulcans"}, {"Y2", "Vulcans"}, {"Kronos", "Humans"}})));
|
||||
auto r3 = u1.colonise({{"Unknown", "Unknown", 0}});
|
||||
assert(r3 == (map<string, string> ({{"Unknown", "Unknown"}})));
|
||||
auto r4 = u1.colonise({});
|
||||
assert(r4 == (map<string, string>({})));
|
||||
|
||||
|
||||
CUniverse u2;
|
||||
|
||||
u2 .addConnection("Earth", "Z1")
|
||||
.addConnection("Earth", "Y1")
|
||||
.addConnection("Earth", "Kronos")
|
||||
.addConnection("Earth", "Vulcan")
|
||||
.addConnection("Vulcan", "Y3")
|
||||
.addConnection("Vulcan", "X1")
|
||||
.addConnection("Kronos", "Z2")
|
||||
.addConnection("Kronos", "X4")
|
||||
.addConnection("Kronos", "Vulcan")
|
||||
.addConnection("Y1", "Y2")
|
||||
.addConnection("Y2", "Y3")
|
||||
.addConnection("Z1", "Z2")
|
||||
.addConnection("X1", "X2")
|
||||
.addConnection("X2", "X3")
|
||||
.addConnection("X1", "X3")
|
||||
.addConnection("X3", "X4")
|
||||
.addConnection("Bajor", "Cassiopea Prime")
|
||||
.optimize();
|
||||
|
||||
auto r5 = u2.colonise({{"Earth", "Humans", 0}, {"Vulcan", "Vulcans", 0}, {"Kronos", "Clingons", 0}, {"Cassiopea Prime", "Cassiopeans", 0}});
|
||||
assert(r5 == (map<string, string> ({{"Earth", "Humans"}, {"Kronos", "Clingons"}, {"Vulcan", "Vulcans"}, {"Cassiopea Prime", "Cassiopeans"}, {"Bajor", "Cassiopeans"}, {"Z1", "Humans"}, {"Z2", "Clingons"}, {"Y1", "Humans"}, {"Y3", "Vulcans"}, {"X1", "Vulcans"}, {"X2", "Vulcans"}, {"X4", "Clingons"}})));
|
||||
|
||||
//Harder tests when != 0 for all colonies
|
||||
CUniverse u3;
|
||||
|
||||
u3 .addConnection("Earth", "Z1")
|
||||
.addConnection("Earth", "Y1")
|
||||
.addConnection("Earth", "Kronos")
|
||||
.addConnection("Earth", "Vulcan")
|
||||
.addConnection("Vulcan", "Y3")
|
||||
.addConnection("Vulcan", "X1")
|
||||
.addConnection("Kronos", "Z2")
|
||||
.addConnection("Kronos", "X4")
|
||||
.addConnection("Y1", "Y2")
|
||||
.addConnection("Y2", "Y3")
|
||||
.addConnection("Z1", "Z2")
|
||||
.addConnection("X1", "X2")
|
||||
.addConnection("X1", "X3")
|
||||
.addConnection("X2", "X3")
|
||||
.addConnection("X3", "X4")
|
||||
.addConnection("Bajor", "Cassiopea Prime")
|
||||
.optimize();
|
||||
|
||||
auto r8 = u3.colonise({{"Earth", "Humans", 1}, {"Vulcan", "Vulcans", 0}, {"Kronos", "Clingons", 2}, {"Cassiopea Prime", "Cassiopeans", 10000}});
|
||||
assert(r8 == (map<string, string> ({{"Earth", "Humans"}, {"Kronos", "Clingons"}, {"Vulcan", "Vulcans"}, {"Y1", "Humans"}, {"Z1", "Humans"}, {"Y3", "Vulcans"}, {"Y2", "Vulcans"}, {"X1", "Vulcans"}, {"X2", "Vulcans"}, {"X3", "Vulcans"}, {"Cassiopea Prime", "Cassiopeans"}, {"Bajor", "Cassiopeans"}})));
|
||||
|
||||
std::cout << "success!\n";
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
#endif /* __PROGTEST__ */
|
||||
Loading…
Reference in New Issue