2 курс. Программирование на C++. Турнир 2, осень 2021

вход курсы / турниры задачи участники консоль модули компиляторы ключевые слова результаты

ru en cn

5787008

Пользователь

Кузьмин Павел Владимирович

Время

20:35<< 16.11.2021 20:36 (27 д. 18:36)

Обработка

20:36 ... 20:36 (5,17 сек.)

Файл

5787008.cpp (CLang++ 5.0.0 + Address Sanitizer)

Турнир

2 курс. Программирование на C++. Турнир 2, осень 2021

Задача

E: GameDatabase

Набор тестов

Ограничения

Время: 1с Память: 4000Мб

Состояние [?]

Принято

код | результаты

[печатать | | ]

≡

#include <map>
#include <set>
#include <string>
#include <iterator>
#include <utility>
#include <vector>
#include <functional>
#include <algorithm>

bool operator>(const GameObject& a, const GameObject& b) {
    return a.id > b.id;
}

template<class T, template<class> class Compare>
class DereferenceCompare {
 public:
    bool operator()(const T* const a, const T* const b) const {
        return comp(*a, *b);
    }
 private:
    Compare<T> comp;
};

using TPos = std::pair<size_t, size_t>;
using TSet = std::set<GameObject*, DereferenceCompare<GameObject,
std::greater>>;

class GameDatabase {
 public:
    GameDatabase() = default;

void Insert(ObjectId id, std::string name, size_t x, size_t y);
    void Remove(ObjectId id);
    std::pair<GameObject, bool> DataById(ObjectId id) const;
    std::vector<GameObject> DataByName(std::string name) const;
    std::vector<GameObject> DataByPosition(size_t x, size_t y) const;
    std::vector<GameObject> Data() const;

private:
    std::map<ObjectId, GameObject, std::greater<ObjectId>> data;
    std::map<TPos, TSet> by_pos;
    std::map<std::string, TSet> by_name;
};

void GameDatabase::Insert(ObjectId id, std::string name, size_t x, size_t y) {
    GameObject temp = {id, name, x, y};
    if (data.find(id) != data.end())
        Remove(id);
    data.insert(std::make_pair(id, temp));
    by_name[name].insert(&data[id]);
    TPos pos = std::make_pair(x, y);
    by_pos[pos].insert(&data[id]);
}

void GameDatabase::Remove(ObjectId id) {
    auto it = data.find(id);
    if (it != data.end()) {
        GameObject obj = it->second;
        by_name[obj.name].erase(&obj);
        TPos pos = std::make_pair(obj.x, obj.y);
        by_pos[pos].erase(&obj);
        data.erase(it);
    }
}

std::pair<GameObject, bool> GameDatabase::DataById(ObjectId id) const {
    auto result = data.find(id);
    return std::make_pair(result->second, result != data.end());
}

std::vector<GameObject> GameDatabase::DataByName(std::string name) const {
    std::vector<GameObject> result;
    if (by_name.find(name) != by_name.end()) {
        TSet container = by_name.find(name)->second;
        for (auto it = container.begin(); it != container.end(); it++) {
            result.push_back(**it);
        }
    }
    return result;
}

std::vector<GameObject> GameDatabase::DataByPosition(size_t x, size_t y) const {
    std::vector<GameObject> result;
    TPos pos = std::make_pair(x, y);
    if (by_pos.find(pos) != by_pos.end()) {
        TSet container = by_pos.find(pos)->second;
        for (auto it = container.begin(); it != container.end(); it++) {
            result.push_back(**it);
        }
    }
    return result;
}

std::vector<GameObject> GameDatabase::Data() const {
    std::vector<GameObject> result;
    for (auto it = data.begin(); it != data.end(); it++)
        result.push_back(it->second);
    return result;
}

(2 980 b)

0.052s 0.010s 9