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

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

ru en cn

5775005

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

Псарев Александр Павлович

Время

00:04<< 15.11.2021 00:04 (25 д. 22:05)

Обработка

00:04 ... 00:05 (4,94 сек.)

Файл

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

Турнир

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

Задача

E: GameDatabase

Набор тестов

Ограничения

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

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

Принято

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

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

≡

#include <vector>
#include <algorithm>
#include <unordered_map>
#include <map>
#include <unordered_set>
#include <set>
#include <string>
#include <utility>
#include <functional>

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

template<class Tp, template<class> class Compare>
class DereferenceCompare {
    Compare<Tp> comp;

public:
    bool operator()(const Tp* const a, const Tp* const b) const {
        return comp(*a, *b);
    }
};

class GameDatabase {
 private:
    std::map<ObjectId, GameObject, std::greater<ObjectId>> obj_id;

std::map<std::pair<size_t, size_t>, std::set<GameObject*, \
    DereferenceCompare<GameObject, std::greater>>> obj_pos;

std::unordered_map<string, std::set<GameObject*, \
    DereferenceCompare<GameObject, std::greater>>> obj_name;

public:
    GameDatabase() = default;

/// вставляет в базу объект с именем [name] и позицией [x, y]
    /// если объект с таким id в базе уже есть, заменяет его новым
    void Insert(ObjectId id, string name, size_t x, size_t y) {
        GameObject object{ id, name, x, y };
        if (obj_id.find(id) != obj_id.end()) {
            Remove(id);
        }
        obj_id.insert({ id, object });
        auto temp = &obj_id.at(id);
        obj_name[name].insert(temp);
        obj_pos[{x, y}].insert(temp);
    }

/// удаляет элемент по id
    /// если такого элемента нет, ничего не делает
    void Remove(ObjectId id) {
        if (obj_id.find(id) != obj_id.end()) {
            GameObject* object = &obj_id.find(id)->second;
            obj_name.find(object->name)->second.erase(object);
            obj_pos.find({ object->x, object->y })->second.erase(object);
            obj_id.erase(obj_id.find(id));
        }
    }

/// возвращает вектор объектов c именем [name]
    /// сортировка по убыванию id
    std::vector<GameObject> DataByName(string name) const {
        std::vector <GameObject> result;
        auto check = obj_name.find(name);
        if (check == obj_name.end()) {
            return result;
        }
        for (auto temp : check->second) {
            result.push_back(*temp);
        }
        return result;
    }

/// возвращает вектор объектов, находящихся в позиции [x, y]
    /// сортировка по убыванию id
    std::vector<GameObject> DataByPosition(size_t x, size_t y) const {
        std::vector <GameObject> result;
        auto check = obj_pos.find({ x, y });
        if (check == obj_pos.end()) {
            return result;
        }
        for (auto temp : check->second) {
            result.push_back(*temp);
        }
        return result;
    }

/// возвращает вектор всех объектов из базы
    /// сортировка по убыванию id
    std::vector<GameObject> Data() const {
        std::vector <GameObject> result;
        for (const auto& obj : obj_id) {
            result.push_back(obj.second);
        }
        return result;
    }
};

(3 318 b)

0.093s 0.009s 9