将 std::map 的数据拆分为多个 std::vectors

Question

我有两个类：InitTable 和InitEntry。 InitTable 包含一个 std::map table 存储条目（汽车）的 ID 和代表该汽车的对象 (InitEntry)。每个InitEntry都有3个成员变量：

std::string ID
double speed
std::string heading

我的目标是首先将所有汽车存储在 std::map table 中，然后遍历该数据结构，并尝试根据共同属性将汽车组织成集群 (std::vector<InitEntry>)： speed & heading.

示例：

假设我们有 6 辆车（id 0 到 8）

car0：id：“0”，速度：22，航向 "N"
car1：id：“1”，速度：26，航向 "N"
car2：id：“2”，速度：28，航向 "W"
car3：id：“3”，速度：12，航向 "E"
car4：id：“4”，速度：10，航向 "E"
car5：id：“5”，速度：45，航向 "S"

为简单起见，在现阶段对我来说，仅根据航向对汽车进行分组就足够了。结果将是：

std::vector clus1 = {car0, car1}
std::vector clus2 = {car2}
std::vector clus3 = {car3, car4}
std::vector clus4 = {car5}

不幸的是，我对 C++ STL 的了解不够，无法理解如何在 C++ 中实现这一点。

InitTable.h:

    #include <InitEntry.h>


    class InitTable {
        public:
            InitTable();
            virtual ~InitTable();

            void clearTable();
            void addEntry(std::string ID, double speed, std::string heading);
            void deleteEntry(std::string ID);
            InitEntry* getEntry(std::string ID);


        protected:
            std::map<std::string, InitEntry*> table;
    };

InitTable.cc:

#include"InitTable.h"

InitTable::InitTable(){}


InitTable::~InitTable()
{
    clearTable();
}


void InitTable::clearTable()
{
    this->table.clear();
}

void InitTable::addEntry(std::string ID, double speed, std::string heading)
{
    InitEntry* newEntry = new InitEntry(ID, speed, heading);


    std::cout<< "server::InitTable: vehicle registered to the init table" << newEntry << endl;

    table.insert(std::make_pair(ID, newEntry));

}



void InitTable::deleteEntry(std::string ID)
{
    InitEntry* ie = getEntry(ID);
    if (ie != NULL)
    {
        table.erase(ID);
        delete ie;
    }
}

InitEntry* InitTable::getEntry(std::string ID)
{
    std::map<std::string, InitEntry*>::iterator it = table.find(ID);

    if (it != table.end())
    {
        return it->second;
    }
    else
    {
        std::cout << "such entry does not exist" << endl;
        return NULL;
    }
}

InitEntry.h:

class InitEntry {
    public:
        virtual ~InitEntry();
        InitEntry(std::string ID, double speed, std::string heading);
        std::string getID();


    protected:
        std::string sumoID;
        double speed;
        std::string heading;

};

InitEntry.cc:

#include "InitEntry.h"


InitEntry::InitEntry(std::string ID, double speed, std::string heading): ID(ID), speed(speed), heading(heading){}


InitEntry::~InitEntry(){}

std::string InitEntry::getID()
{
    return this->ID;
}

编辑 1：添加额外描述（应@TomaszLewowski 的要求）。

是的，我的目标是按航向然后根据速度将车辆组织成集群。因此，最初会有一大群车辆朝某个方向行驶，之后需要根据速度将其分成更多的集群。比方说：车辆行驶方向"north"，速度：0 - 20 ...速度速度：40 - 50 ...等

Answer 1

考虑将您的 std::map<std::string, InitEntry*> table; 更改为：

std::vector<InitEntry> table;

然后您需要将 *Entry 方法更改为：

void InitTable::addEntry(std::string ID, double speed, std::string heading)
{
    table.push_back(InitEntry(ID, speed, heading));
    std::cout<< "server::InitTable: vehicle registered to the init table" << table.back() << endl;
}

void InitTable::deleteEntry(std::string ID)
{
    auto it = remove_if(table.begin(), table.end(), [&](const auto& i){return i.getID() == ID;});
    table.resize(distance(table.begin(), it));
}

InitEntry* InitTable::getEntry(std::string ID)
{
    auto it = find_if(table.begin(), table.end(), [&](const auto& i){return i.getID() == ID;});

    if (it != table.end())
    {
        return it->second;
    }
    else
    {
        std::cout << "such entry does not exist" << endl;
        return NULL;
    }
}

要排序 table，您需要决定排序依据。

按ID排序：std::sort(table.begin(), table.end(), [](const auto& first, const auto& second){return first.getID() < second.getID();});
按速度排序：std::sort(table.begin(), table.end(), [](const auto& first, const auto& second){return first.getSpeed() < second.getSpeed();});
按标题排序：std::sort(table.begin(), table.end(), [](const auto& first, const auto& second){return first.getHeading() < second.getHeading();});

显然，您需要添加吸气剂才能使这些工作正常进行。

Answer 2

如果您不想遍历所有元素（我想您也不想），那么唯一的选择就是将这些组保留在某个地方，并且在主 table 中只对它们进行一些索引.这看起来大致像这样：

#include <string>
#include <vector>
#include <map>
#include <cassert>
#include <algorithm>

typedef std::string ID;
typedef std::string Heading;
typedef double Speed;

struct Entry
{
    ID id;
    Speed speed;
    Heading heading;
};

class EntryProxy
{
public:
    EntryProxy(Entry* target) : entry(target)
    {}

    ID getId()
    {
        return entry->id;
    }

    Speed getSpeed()
    {
        return entry->speed;
    }

    Heading getHeading()
    {
        return entry->heading;
    }

private:
    Entry* entry;
};

class InitTable
{
public:
    const std::vector<EntryProxy>& getSameHeading(std::string id)
    {
        return groupedByHeadings.at(entries.at(id).heading);
    }

    const std::vector<EntryProxy>& getSimilarSpeed(std::string id)
    {
        return groupedBySpeed.at(calculateSpeedGroupIndex(entries.at(id).speed));
    }

    void addEntry(ID id, Speed speed, Heading heading)
    {
        Entry e{ id, speed, heading };
        auto record = entries.insert(std::make_pair(id, e)); // pair<iterator, bool>

        groupedByHeadings[record.first->second.heading].push_back(EntryProxy(&record.first->second));
        groupedBySpeed[calculateSpeedGroupIndex(record.first->second.speed)].push_back(EntryProxy(&record.first->second));
    }

    void deleteEntry(ID id)
    {
        auto entry = entries.find(id);
        assert(entry != entries.end());

        auto currentEntryHeadings = groupedByHeadings[entry->second.heading];
        currentEntryHeadings.erase(std::find_if(currentEntryHeadings.begin(), currentEntryHeadings.end(), [&entry](EntryProxy p){return p.getId() == entry->second.id; }));

        auto currentEntrySpeeds = groupedBySpeed[calculateSpeedGroupIndex(entry->second.speed)];
        currentEntrySpeeds.erase(std::find_if(currentEntrySpeeds.begin(), currentEntrySpeeds.end(), [&entry](EntryProxy p){return p.getId() == entry->second.id; }));

        entries.erase(id);
    }

    EntryProxy getEntry(ID id)
    {
        return EntryProxy(&entries.at(id));
    }

private:
    typedef int SpeedGroupIndex;

    SpeedGroupIndex calculateSpeedGroupIndex(Speed s)
    {
        // you may prefer a different implementation here

        return s / 10;
    }

    std::map<ID, Entry> entries;
    std::map<Heading, std::vector<EntryProxy> > groupedByHeadings;
    std::map<SpeedGroupIndex, std::vector<EntryProxy> > groupedBySpeed;
};

按标题分组非常明显，因为它只需要一张简单的地图。然而，当谈到速度时，它有点困难——你需要描述分组的要求（通过实现函数 getSpeedGroupId），因为双打不应该用作地图索引。

如果您更喜欢更安全的内存管理方法，您可以用 Entry* 代替 EntryProxy 中的 std::shared_ptr<Entry>，用 Entry 代替 std::shared_ptr<Entry> InitTable::entries。在这种情况下，您还需要将创建从 Entry e{id, speed, heading 更改为 auto e = std::make_shared<Entry>(id, speed, heading).

如果删除对您来说太复杂，您可以将 EntryProxy 结构保留在列表中，并在 entries 映射中保留对特定迭代器的引用。

您可能也不喜欢这样的事实，即组是 EntryProxy 而单个元素是 Entry - 在这种情况下，您可以将 getEntry 也设为 return EntryProxy.

将 std::map 的数据拆分为多个 std::vectors

splitting data of a std::map into multiple std::vectors

c++

dictionary

stl

vector

c++98