2021-07-13

用queue实现广度遍历算法

思路

​ 在一个二维表树里面，一般来说根节点的id为0，剩下的节点都有相应的id，要从根节点开始查找直到全部查完，首先向队列里面放入根节点的id，查找此id的所有子节点id放入队列中，将此id弹出队列，依次对队列中的所有id进行此操作。直到最后队列变为空。这棵树就遍历完成了。

代码

int getAllChildlist(string& PID, int& navitemType, int page, int pageSize, vector<string>& CID)
{
    queue<string> ids;
    vector<string> nextLevelIds;

    getNextLevelID(navitemType, page, pageSize, PID, nextLevelIds);
    for(string item : nextLevelIds)
    {
        ids.push(item);
        CID.push_back(item);
    }

    while (ids.size())
    {
        string id = ids.front();
        ids.pop();
        getNextLevelID(navitemType, page, pageSize, id, nextLevelIds);
        for(string item : nextLevelIds)
        {
            ids.push(item);
            CID.push_back(item);
        }
        nextLevelIds.clear();
    }
    return 0;
}

int getNextLevelID(int& navitemType, int page, int pageSize, string &id, vector<string> &nextLevelIds)
{
    vector<map<string, string> > records;
    vector<string> fields;
    map<string, string> where;
    fields.push_back(D_NAVITEMTABLE_NAVITEMID);
    if (id != "")
    {
        where[D_NAVITEMTABLE_PID] = id;
    }

    if (navitemType)
    {
        where[D_NAVITEMTABLE_NAVITEMTYPE] = CUtils::int2str(navitemType);
    }
    where["order by"] = CUtils::stringFormat("%s limit %s,%s", D_NAVITEMTABLE_NAVITEMID, CUtils::int2str((page - 1) * pageSize).c_str(), CUtils::int2str(pageSize).c_str());
    int ret = rwdb->query(D_NAVITEMTABLE_TABLENAME, records, fields, where);
    if (ret)
    {
        ERR_INFO(STR_LOGFILE_CYGCOMMONNAVITEM, "CNavitemL::getNavitemList failed");
        return NAVITEM_RWDB_ERROR;
    }

    nextLevelIds.clear();
    for (const map<string, string>& record : records)
    {
        nextLevelIds.push_back(record.at(D_NAVITEMTABLE_NAVITEMID));
    }
    return 0;
}