pkg

pkg.cpp

@@ -0,0 +1,171 @@
+#include <iostream>
+#include <cassert>
+#include <vector>
+#include <set>
+#include <map>
+#include <string>
+#include <unordered_map>
+#include <memory>
+#include <unordered_set>
+#include <queue>
+#include <list>
+#include <sstream>
+#include <cstring>
+ 
+ #include <stack>
+
+using namespace std;
+
+class CPkg{
+public:
+    CPkg( std::string n )
+    : _name(std::move(n)) {}
+    CPkg& addDep( std::string n ) {
+        _deps.emplace_back( std::move(n) );
+        return *this;
+    }
+    bool operator<( const CPkg& oth ) const {
+        return _name < oth._name;
+    }
+
+    std::string _name;
+    std::vector< std::string > _deps;
+private:
+ 
+};
+
+class CPkgSys{
+public:
+    CPkgSys& addPkg(const CPkg& p) {
+        packages[p._name] = p._deps;
+        return *this;
+    }
+
+    std::set<std::string> install(std::list<std::string> what) {
+        std::set<std::string> d = installed;
+        std::queue<std::string> q;
+        std::set<std::string> r;
+
+        for (const auto& i : what) {
+            q.push(i);
+            if (d.emplace(i).second)
+                r.emplace(i);
+        }
+
+        while (q.size()) {
+            auto i = q.front();
+            q.pop();
+            auto it = packages.find(i);
+
+            if (it == packages.end())
+                throw std::invalid_argument("Package not found.");
+            
+            for ( const auto& j : it->second ) {
+                if (d.emplace(j).second) {
+                    r.emplace(j);
+                    q.push(j);
+                }
+            }
+        }
+        std::swap( installed, d );
+        return r;
+    }
+
+private:
+    std::unordered_map<std::string, std::vector<std::string>> packages;
+    std::set<std::string> installed;
+};
+
+int main(void){
+    CPkgSys s;
+    stringstream ss;
+    s.addPkg(CPkg("ssh").addDep("sudo").addDep("apt"))
+    .addPkg(CPkg("sudo").addDep("git").addDep("c++"));
+    s.addPkg(CPkg("apt"))
+    .addPkg(CPkg("c++").addDep("c").addDep("asm").addDep("fortran"));
+    s.addPkg(CPkg("git"))
+    .addPkg(CPkg("c").addDep("kekw"))
+    .addPkg(CPkg("kekw"))
+    .addPkg(CPkg("asm"))
+    .addPkg(CPkg("fortran"));
+    s.addPkg(CPkg("python").addDep("bash").addDep("sadge"))
+    .addPkg(CPkg("karel").addDep("python"))
+    .addPkg(CPkg("bash").addDep("sadge"))
+    .addPkg(CPkg("sadge"))
+    .addPkg(CPkg("cython").addDep("dev"));
+    s.addPkg(CPkg("perl"));
+ 
+    // ss << s;
+    // assert(ss.str() == "");
+    ss.clear();
+    ss.str("");
+ 
+    set<string> t1 = s.install(list<string> {"sudo"});
+    assert(t1 == (set<string> {"asm", "c", "c++", "fortran", "git", "kekw", "sudo"}));
+    set<string> t2 = s.install(list<string> {"ssh", "c++"});
+    assert(t2 == (set<string> {"apt", "ssh"}));
+ 
+    // ss << s;
+    // assert(ss.str() == "apt, asm, c, c++, fortran, git, kekw, ssh, sudo");
+    // ss.clear();
+    // ss.str("");
+ 
+    try{
+        set<string> e = s.install(list<string> {"karel", "cython"});
+        assert("Sem ses nemel dostat debilku" == nullptr);
+    }
+    catch(const invalid_argument & e){
+        assert(strcmp("Package not found.", e.what()) == 0);
+    }
+    set<string> t3 = s.install(list<string> {"karel", "fortran", "git"});
+    assert(t3 == (set<string> {"bash", "karel", "python", "sadge"}));
+ 
+    s.addPkg(CPkg("java").addDep("utils"))
+    .addPkg(CPkg("utils").addDep("VB"))
+    .addPkg(CPkg("VB").addDep("java"));
+ 
+    set<string> t4 = s.install(list<string> {"java", "perl"});
+    assert(t4 == (set<string> {"VB", "java", "perl", "utils"}));
+ 
+    // ss << s;
+    // assert(ss.str() == "VB, apt, asm, bash, c, c++, fortran, git, java, karel, kekw, perl, python, sadge, ssh, sudo, utils");
+    // ss.clear();
+    // ss.str("");
+ 
+    CPkgSys k;
+    k.addPkg(CPkg("ssh").addDep("sudo").addDep("apt"))
+    .addPkg(CPkg("sudo").addDep("git"));
+    k.addPkg(CPkg("apt"));
+    k.addPkg(CPkg("git"))
+    .addPkg(CPkg("c").addDep("kekw"))
+    .addPkg(CPkg("kekw"))
+    .addPkg(CPkg("asm"))
+    .addPkg(CPkg("fortran"));
+    k.addPkg(CPkg("python").addDep("bash").addDep("sadge"))
+    .addPkg(CPkg("karel").addDep("python"))
+    .addPkg(CPkg("bash").addDep("sadge"))
+    .addPkg(CPkg("sadge"));
+    k.addPkg(CPkg("perl").addDep("no"));
+ 
+    set<string> t5 = k.install(list<string> {"asm"});
+    assert(t5 == (set<string> {"asm"}));
+    set<string> t6 = k.install(list<string> {"python", "ssh"});
+    assert(t6 == (set<string> {"apt", "bash", "git", "python", "sadge", "ssh", "sudo"}));
+ 
+    try{
+        set<string> t7 = k.install(list<string> {"perl", "c"});
+        assert("Sem ses nemel dostat debilku" == nullptr);
+    }
+    catch(const invalid_argument & e){
+        assert(strcmp("Package not found.", e.what()) == 0);
+    }
+    set<string> t8 = k.install(list<string> {"c", "ssh", "karel"});
+ 
+    assert(t8 == (set<string> {"c", "karel", "kekw"}));
+ 
+    // ss << k;
+    // assert(ss.str() == "apt, asm, bash, c, git, karel, kekw, python, sadge, ssh, sudo");
+    // ss.clear();
+    // ss.str("");
+    std::cout << "success!\n";
+}