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

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

ru en cn

5732146

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

Травка Евгения Александровна

Время

11.11.2021 18:50 (22 д. 16:50)

Обработка

18:50 ... 18:50 (6,51 сек.)

Файл

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

Турнир

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

Задача

E: GameDatabase

Набор тестов

Ограничения

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

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

Принято

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

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

≡

#pragma once
#include <string>
#include <vector>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#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 {
 public:
  GameDatabase() = default;
  void Insert(ObjectId id, std::string name, size_t x, size_t y);
  void Remove(ObjectId id);
  vector<GameObject> DataByName(std::string name) const;
  vector<GameObject> DataByPosition(size_t x, size_t y) const;
  vector<GameObject> Data() const;

private:
  std::map<ObjectId, GameObject, std::greater<ObjectId>> id_;
  std::map<std::pair<size_t, size_t>, std::set<GameObject*, \
  DereferenceCompare<GameObject, std::greater>>> position_;
  std::unordered_map<std::string, std::set<GameObject*, \
  DereferenceCompare<GameObject, std::greater>>> name_;
};

void GameDatabase::Insert(ObjectId id, std::string name, size_t x, size_t y) {
  GameObject tmp{id, name, x, y};
  if (id_.find(id) != id_.end()) Remove(id);
  id_.insert({id, tmp});
  auto tmp1 = &id_.at(id);
  position_[{x, y}].insert(tmp1);
  name_[name].insert(tmp1);
}

void GameDatabase::Remove(ObjectId id) {
  if (id_.find(id) != id_.end()) {
    GameObject* tmp = &id_.find(id)->second;
    (position_.find({tmp->x, tmp->y}))->second.erase(tmp);
    (name_.find({tmp->name}))->second.erase(tmp);
    id_.erase(id_.find(id));
  }
}

std::vector<GameObject> GameDatabase::DataByName(std::string name) const {
  std::vector<GameObject> result;
  auto iterator = name_.find(name);
  if (iterator == name_.end()) return result;
  for (auto i : iterator->second) {
    result.push_back(*i);
  }
  return result;
}

std::vector<GameObject> GameDatabase::DataByPosition(size_t x, size_t y) const {
  std::vector<GameObject> result;
  auto iterator = position_.find({x, y});
  if (iterator == position_.end()) return result;
  for (auto i : iterator->second) {
    result.push_back(*i);
  }
  return result;
}

std::vector<GameObject> GameDatabase::Data() const {
  std::vector<GameObject> result;
  for (const auto& i : id_) {
    result.push_back(i.second);
  }
  return result;
}

(2 362 b)

0.085s 0.015s 9